python - numpy 中的矩阵乘法

Question

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

如果我进行矩阵乘法并且元素为 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]])