python - 整数のリストへのバイナリnumpy配列？

Question

バイナリ配列があり、それを整数のリストに変換したいと思います。ここで、各intは配列の行です。

例えば：

from numpy import *
a = array([[1, 1, 0, 0], [0, 1, 0, 0], [0, 1, 1, 1], [1, 1, 1, 1]])

に変換aしたいのですが[12, 4, 7, 15]。

Answer 1

@SteveTjoaの答えは問題ありませんが、キックの場合は、ここに厄介なワンライナーがあります。

In [19]: a
Out[19]: 
array([[1, 1, 0, 0],
       [0, 1, 0, 0],
       [0, 1, 1, 1],
       [1, 1, 1, 1]])

In [20]: a.dot(1 << arange(a.shape[-1] - 1, -1, -1))
Out[20]: array([12,  4,  7, 15])

（arangeですnumpy.arange。）

arangeビットの順序が逆の場合は、 ：によって生成される値の順序を変更します。

In [25]: a.dot(1 << arange(a.shape[-1]))
Out[25]: array([ 3,  2, 14, 15])

Answer 2

私はかつてここで同様の質問をしました。これが私の答えで、あなたの質問に適合しました：

def bool2int(x):
    y = 0
    for i,j in enumerate(x):
        y += j<<i
    return y

In [20]: a
Out[20]: 
array([[1, 1, 0, 0],
       [0, 1, 0, 0],
       [0, 1, 1, 1],
       [1, 1, 1, 1]])

In [21]: [bool2int(x[::-1]) for x in a]
Out[21]: [12, 4, 7, 15]

Answer 3

numpy内で直接これを行うこともできます：

from numpy import *
a = array([[1, 1, 0, 0], [0, 1, 0, 0], [0, 1, 1, 1], [1, 1, 1, 1]])

b2i = 2**arange(a.shape[0]-1, -1, -1)

result = (a*b2i).sum(axis=1)  #[12  4  7 15]

Answer 4

ビット単位の数学を直接操作するのが好きな場合は、これでかなりうまくいくはずです。

def bits2int(a, axis=-1):
    return np.right_shift(np.packbits(a, axis=axis), 8 - a.shape[axis]).squeeze()

bits2int(a)
Out: array([12,  4,  7, 15], dtype=uint8)

Answer 5

もう1つ：

def row_bits2int(arr):
    n = arr.shape[1]  # number of columns

    # shift the bits of the first column to the left by n - 1
    a = arr[:, 0] << n - 1  

    for j in range(1, n):
        # "overlay" with the shifted bits of the next column
        a |= arr[:, j] << n - 1 - j  
    return a