将 [0-1] 列表快速转为二进制表示

Question

我正在尝试将矩阵的 [0-1] 行转换为数字表示（二进制等效），我拥有的代码如下：

import numpy as np

def generate_binary_matrix(matrix):
    result = []
    for i in matrix:
        val = '0b' + ''.join([str(x) for x in i])
        result.append(int(val, 2))
    result = np.array(result)
    return result

initial_matrix = np.array([[0, 1, 0], [1, 0, 0], [0, 0, 1]])
result = generate_binary_matrix(initial_matrix )
print(result)

此代码有效，但速度很慢，有谁知道如何以更快的方式完成？

Answer 1

您可以仅使用算术将 0/1 列表转换为二进制，这应该更快：

from functools import reduce
b = reduce(lambda r, x: 2*r + x, i)

Answer 2

假设您的矩阵 numpy 数组是 A 与 m 行和 n 列。 使用nelements 创建一个b 向量：

b = np.power(2, np.arange(n))[::-1]

那么你的答案是A @ b

例子：

import numpy as np
A = np.array([[0, 0, 1], [1, 0, 1]])
n = A.shape[1]
b = np.power(2, np.arange(n))[::-1]
print(A @ b)  # --> [1 5]

更新 - 我颠倒了 b，因为 MSB (2^n-1) 是 A[:,0] + 幂参数被错误地翻转 + 添加示例。