从另一个数组中添加值以计算numpy中的总和

Question

是否有一种方法可以在numpy中创建这种类型的累计和，而不使用for循环？

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

我对如何使用cumsum函数以便在将数组x的值相加时计算数组y的累加和感到困惑。

Answer 1

我知道了！

这是执行此操作的代码：

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