二进制numpy数组到整数列表？

小编典典

二进制numpy数组到整数列表？

python

我有一个二进制数组，我想将其转换为整数列表，其中每个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]。

阅读 215

2020-12-20

共1个答案

小编典典

我曾经在这里问过类似的问题。这是我的回答，适合您的问题：

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]

2020-12-20