python - numpy中具有不同形状的高效乘法矩阵

Question

我有两个矩阵 A 和 B:

>>>A
[[1,1,1]
[1,1,1]]
>>>B
[2, 3]

我想将它们有效地相乘以获得这样的结果:

>>> A*B
[[2,2,2]
[3,3,3]]

我正在寻找比循环迭代更有效的解决方案。谁能帮忙？

Answer 1

您可以使用 np.multiply将元素与广播相乘:

A = np.array([[1,1,1],
              [1,1,1]])

B = np.array([2, 3])

res = np.multiply(A, B[:, None])

print(res)

array([[2, 2, 2],
       [3, 3, 3]])