运行一维NumPy数组的累积和
我有一个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的详细说明。