Для поэлементного умножения двух массивов NumPy с одинаковым измерением просто 'A * B'.
In [1]: a = arange(50)
In [2]: b = ones(50) * 2
In [3]: a
Out[3]:
array([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33,
34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49])
In [4]: b
Out[4]:
array([ 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2.,
2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2.,
2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2.,
2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2.])
In [5]: a * b
Out[5]:
array([ 0., 2., 4., 6., 8., 10., 12., 14., 16., 18., 20.,
22., 24., 26., 28., 30., 32., 34., 36., 38., 40., 42.,
44., 46., 48., 50., 52., 54., 56., 58., 60., 62., 64.,
66., 68., 70., 72., 74., 76., 78., 80., 82., 84., 86.,
88., 90., 92., 94., 96., 98.])
In [6]: (a * b).sum()
Out[6]: 2450.0
Если вы можете писать вещи в терминах умножения матриц, вы можете использовать dot()
:
In [7]: A = arange(25).reshape((5,5))
In [8]: X = arange(5)
In [9]: A
Out[9]:
array([[ 0, 1, 2, 3, 4],
[ 5, 6, 7, 8, 9],
[10, 11, 12, 13, 14],
[15, 16, 17, 18, 19],
[20, 21, 22, 23, 24]])
In [12]: dot(A, X) # Sum_j A[i,j] * X[j] for all i
Out[12]: array([ 30, 80, 130, 180, 230])
Это должно начать вас.