Я должен запрограммировать функцию suma_a_cada_columna(mat, vec)
, которая, учитывая матрицу и вектор, возвращает матрицу, полученную путем добавления вектора в каждый столбец.
Количество рядов матов соответствует размеру соседнего.
suma_a_cada_columna(np.array([[0, 0, 0], [0, 1, 0], [2, 1, 0]]), np.array([1, 2, 3]))
Результат:
array([[1, 1, 1],[2, 3, 2],[5, 4, 3]])
Не, если существует более прямой / работоспособный способ это делать, но ты можешь transponer массив или вектор и использовать сумму в контексте NumPy:
import numpy as np
def suma_a_cada_columna(matriz, vector):
return (matriz.transpose() + vector).transpose()
Таким образом, если мы звоним в функцию:
suma_a_cada_columna(np.array([[0, 0, 0], [0, 1, 0], [2, 1, 0]]), np.array([1, 2, 3])
Мы получаем как вывод:
numpy.array([[1 1 1]
[2 3 2]
[5 4 3]])