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

时间:2020-08-17 11:17:37

标签: python numpy matrix-multiplication

它可能以前曾被提及过,但我找不到答案。可以说我有两个条件:

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循环来做到这一点,但是有没有一个功能可以满足我的需要,我需要它比循环更快。 谢谢

2 个答案:

答案 0 :(得分: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]])

答案 1 :(得分:0)

您可以只使用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]]