将具有相同索引的两个矩阵的元素相乘

Question

以前可能有人对此感到困惑，但我找不到答案。假设我有两个妈妈：

A = [[1,2,3],
     [4,5,6]]
B = [[1,2,3]
     [4,5,6]]

我想把它们逐元素相乘，例如索引[0,0]形式A的元素与B的[0,0]相乘，[0,1]A与[0,1]B相乘，。。。等 最后我会得到如下矩阵：

C = [[1,4,9]
     [16,25,36]]

我知道我可以用for循环来做，但是有没有一个函数可以帮我做到这一点，我需要它比循环快。 谢谢

Answer 1

只需使用numpy.multiply

A = np.array([[1,2,3],
     [4,5,6]])


B = np.array([[1,2,3],
     [4,5,6]])

np.multiply(A,B)

array([[ 1,  4,  9],
       [16, 25, 36]])

Answer 2

您可以只使用numpy乘法

A = np.array([[1,2,3],[4,5,6]])
B = np.array([[1,2,3],[4,5,6]])
C = A*B
Out:
[[ 1  4  9]
 [16 25 36]]