python - 如何在Python中将二进制数字行堆栈转换为值列？

Question

这个问题已经有答案了:

Binary numpy array to list of integers? (5 个回答)

已关闭 5 年前。

假设我有

>>> np.array([[0,0],[0,1],[1,0],[1,1]])
array([[0, 0],
       [0, 1],
       [1, 0],
       [1, 1]])

这是一个矩阵，其中的每一行都可以看作某个数字的二进制表示，所以它是

>>> np.array([[0],[1],[2],[3]])
array([[0],
       [1],
       [2],
       [3]])

如何在Python中以最短的方式计算这个转换？

Answer 1

您可以将矩阵乘法与np.dot一起使用使用适当的 2 次方数字对每列进行元素级缩放，然后对每行进行求和归约，这导致我们有几种方法 -

a.dot(2**np.arange(a.shape[1]-1,-1,-1))
a[:,::-1].dot(2**np.arange(a.shape[1]))
a.dot(1 << np.arange(a.shape[1])[::-1])

示例运行 -

In [557]: a = np.array([[0,0],[0,1],[1,0],[1,1]])

In [558]: a.dot(2**np.arange(a.shape[1]-1,-1,-1))
Out[558]: array([0, 1, 2, 3])

In [559]: a[:,::-1].dot(2**np.arange(a.shape[1]))
Out[559]: array([0, 1, 2, 3])

In [566]: a.dot(1 << np.arange(a.shape[1])[::-1])
Out[566]: array([0, 1, 2, 3])