Python numpy unpackbits用法及代碼示例

本文簡要介紹 python 語言中 numpy.unpackbits 的用法。

用法:

numpy.unpackbits(a, /, axis=None, count=None, bitorder='big')

將 uint8 數組的元素解包為二進製值輸出數組。

的每個元素a表示應解壓縮為二進製值輸出數組的 bit-field。輸出數組的形狀是一維的(如果軸是None)或與輸入數組相同的形狀，並沿指定的軸進行解包。

參數：

a： ndarray, uint8 類型

輸入數組。

axis： 整數，可選

完成bit-unpacking 的維度。 None 意味著解壓縮展平的數組。

count： int 或無，可選

要解壓的元素數量軸，作為消除打包大小不是八的倍數的影響的一種方式。非負數表示僅解包數數位。負數意味著從末尾修剪掉那麽多位。None表示解壓整個數組(默認)。大於可用位數的計數將向輸出添加零填充。負計數不得超過可用位數。

bitorder： {‘big’, ‘little’}，可選

返回位的順序。 ‘big’ 將模仿 bin(val)，3 = 0b00000011 => [0, 0, 0, 0, 0, 0, 1, 1]，‘little’ 將順序顛倒到 [1, 1, 0, 0, 0, 0, 0, 0]。默認為‘big’。

返回：

unpacked： ndarray, uint8 類型

元素是二進製值(0 或 1)。

例子：

>>> a = np.array([[2], [7], [23]], dtype=np.uint8)
>>> a
array([[ 2],
       [ 7],
       [23]], dtype=uint8)
>>> b = np.unpackbits(a, axis=1)
>>> b
array([[0, 0, 0, 0, 0, 0, 1, 0],
       [0, 0, 0, 0, 0, 1, 1, 1],
       [0, 0, 0, 1, 0, 1, 1, 1]], dtype=uint8)
>>> c = np.unpackbits(a, axis=1, count=-3)
>>> c
array([[0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0],
       [0, 0, 0, 1, 0]], dtype=uint8)

>>> p = np.packbits(b, axis=0)
>>> np.unpackbits(p, axis=0)
array([[0, 0, 0, 0, 0, 0, 1, 0],
       [0, 0, 0, 0, 0, 1, 1, 1],
       [0, 0, 0, 1, 0, 1, 1, 1],
       [0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 0, 0, 0],
       [0, 0, 0, 0, 0, 0, 0, 0]], dtype=uint8)
>>> np.array_equal(b, np.unpackbits(p, axis=0, count=b.shape[0]))
True