如何从矩阵中每行的矢量化和与另一个矩阵实现矩阵

Posted 2021-03-29

假设我们有A作为A=[[0,0,0],[1,1,1]]和B=[[1,1,1],[2,2,2],[3,3,3]]我如何编写一个矢量化实现，在numpy中没有for循环来计算矩阵C为C=[[6,6,6],[7,7,7]]？通过将A中的每一行与B中的所有行相加来创建C.

答案

做点什么

a = np.array([[0,0,0],[1,1,1]])
b = np.array([[1,1,1],[2,2,2],[3,3,3]])

c = b.sum(axis=0) + a

print(c)

产量

array([[6, 6, 6],
       [7, 7, 7]])