Python: умножение фиксированного массива на второй элементный массив

Question

Пример кода:

import numpy as np
a = np.arange(1,11)
b = np.arange(1,11)
b[:] = 0
b[3] = 10
b[4] = 10
print(a, b)
[ 1  2  3  4  5  6  7  8  9 10] [ 0  0  0 10 10  0  0  0  0  0]

Я пытаюсь умножить b на поэлементный a-массив, чтобы мой результирующий массив был следующим:

[0 0 10 30 50 70 90 110 130 150]

Любая помощь будет с благодарностью

Answer 1

Похоже, вы хотите convolution обоих массивов:

np.convolve(a,b)[:len(a)+1]
# array([  0,   0,   0,  10,  30,  50,  70,  90, 110, 130, 150])

Answer 2

Поэлементное умножение b на a даст вам [0,0,40,50,0,0,0,0,0,0], а не то, что вы указали.