Как получить матрицу из векторизованной суммы каждой строки в матрице с другой матрицей

Question

Предположим, у нас есть A как A=[[0,0,0],[1,1,1]] и B=[[1,1,1],[2,2,2],[3,3,3]] как я могу написать векторизованную реализацию без циклов for для numpy, чтобы вычислить матрицу 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]])