二進制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

我曾經問過類似的問題here。這裏是my answer，適合你的問題：

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 2

你也可以做到這一點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 3

@ SteveTjoa的答案是好的，但踢，這裏有一個numpy的一行代碼：

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。）

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

還有一句：

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