Добавьте значения из другого массива для вычисления суммы в numpy - PullRequest
0 голосов
/ 29 марта 2020

Есть ли способ создания этого типа накопительной суммы в numpy без использования для l oop?

y[0] = x[0]
y[1] = y[0] + x[1]
y[2] = y[1] + x[2]

Я не совсем понимаю, как использовать функцию cumsum для того, чтобы вычислить совокупную сумму массива y при добавлении значений из массива x.

1 Ответ

0 голосов
/ 29 марта 2020

Я понял это!

Вот код для этого:

x = np.arange(0, 10)
y = np.zeros(10)
z = x + y
cum_sum = np.cumsum(z)
...