numpy中的矩阵乘法

Question

我在 numpy 中有 2 个布尔矩阵，并使用 .dot() 函数将它们相乘，得到的结果是一个布尔矩阵。

如果我在做矩阵乘法并且元素是 1 或 0，有没有办法在乘法过程中得到各个元素的乘积之和？

即结果矩阵中的元素应该是 0 或非零整数。

提前致谢。

Answer 1

使用astype 转换为int。

演示：

>>> import numpy as np
>>> np.random.seed(5)
>>> a = np.random.random([3,3]) > 0.5
>>> b = np.random.random([3,3]) > 0.5

现在 a, b 是 dtype bool 的数组：

>>> a
array([[False,  True, False],
       [ True, False,  True],
       [ True,  True, False]], dtype=bool)

将它们作为整数相乘：

>>> np.dot(a.astype(np.int), b.astype(np.int))
array([[0, 0, 1],
       [0, 0, 1],
       [0, 0, 2]])