运行一维NumPy数组的累积和

13 投票
2 回答
7987 浏览
提问于 2025-04-16 16:29

我有一个numpy数组,内容是:

x=np.array([1,2,3,4])

我想创建另一个numpy数组y,它是x的累积和,也就是说:

y=np.array([1,3,6,10])

有什么好的方法可以做到这一点吗?

2 个回答

8

另一种选择是:

y = np.add.accumulate(x)

虽然文档上说这两者是等价的,但实际上这个方法通常比np.cumsum要快。

27
y = np.cumsum(x)

请查看这个链接:http://docs.scipy.org/doc/numpy/reference/generated/numpy.cumsum.html,这里有关于numpy.cumsum的详细说明。

撰写回答