如何用另一个矩阵从矩阵中每一行的矢量和得出矩阵

Question

假设我们将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]]？ C是通过将A中的每一行与B中的所有行相加而创建的。

Answer 1

做类似的事情

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]])