python - 如何从矩阵中每一行的向量化总和与另一个矩阵实现矩阵

Question

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

标签： pythonnumpymatrixdata-science

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