numpy 的使用 uint8 变成 bit 流

背景：对于 numpy 来说，我们使用 uint8 类型，并没有 bit 类型。因此，我抽取了一个函数，使用 unit8 类型，存储bit 类型的数据。代码如下：``python#!/usr/bin/pythonimport numpy as npdef convert_binary_to_bitmap(binary_im):“”"将 uint8 类型的 binary 变成 bitmap “”"h

comedate

1857人浏览 · 2021-08-09 22:19:25

comedate · 2021-08-09 22:19:25 发布

背景：

对于 numpy 来说，我们使用 uint8 类型，并没有 bit 类型。
因此，我抽取了一个函数，使用 unit8 类型，存储bit 类型的数据。

代码如下：


#!/usr/bin/python
import numpy as np

def convert_binary_to_bitmap(binary_im):
    """  将 uint8 类型的 binary 变成 bitmap """
    height = binary_im.shape[0]
    width = binary_im.shape[1]

    # binary image -> bitmap image
    flatten_im = binary_im.flatten()
    bitmap_im = np.zeros((int(width * height / 8)), dtype=np.uint8)
    for idx in list(range(0, flatten_im.shape[0], 8)):
        value = np.uint8(0)
        for idx_inner in range(8):
            value += flatten_im[idx + idx_inner] * 2 ** idx_inner
        bitmap_im[int(idx / 8)] = value
    return width, height, bitmap_im


if __name__ == '__main__':
    binary_im = np.eye(8, dtype=np.uint8)
    print(binary_im)

    _, _, bitmap = convert_binary_to_bitmap(binary_im)
    print(bitmap)
``

### 验证结果：
``
[[1 0 0 0 0 0 0 0]
 [0 1 0 0 0 0 0 0]
 [0 0 1 0 0 0 0 0]
 [0 0 0 1 0 0 0 0]
 [0 0 0 0 1 0 0 0]
 [0 0 0 0 0 1 0 0]
 [0 0 0 0 0 0 1 0]
 [0 0 0 0 0 0 0 1]]
 
[  1   2   4   8  16  32  64 128]
``