Умножьте 2 numpy массивы и верните сумму для каждого измерения

Question

Здравствуйте, моё название может быть немного не так, поэтому позвольте мне объяснить это просто. У меня есть следующие 2 np массива:

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

A состоит из 3 «списков». Первое значение в каждом списке умножается на значение первого числа в b. Второе число в каждом списке, умноженное на второе значение в b et c.

Требуемая калькуляция ниже для желаемого выхода:

[1 * 3 + 2 * 2], [1 * 1 + 4 * 2], [1 * 12 + 4 * 2], что означает:

[7, 9, 20]

Заранее спасибо.

Answer 1

a = np.array([[3,2],[1,4],[12,4]])
b = np.array([1,2])
np.sum(a * b, axis=1)

выход

array([ 7,  9, 20])

Используется numpy вещание