如果你要对像这样的数组做大量的数值工作,我建议使用numpy
,它带有一个累加和函数cumsum
:
import numpy as np
a = [4,6,12]
np.cumsum(a)
#array([4, 10, 22])
在这种情况下,Numpy通常比纯python更快,请参阅与@Ashwini的accumu比较:
In [136]: timeit list(accumu(range(1000)))
10000 loops, best of 3: 161 us per loop
In [137]: timeit list(accumu(xrange(1000)))
10000 loops, best of 3: 147 us per loop
In [138]: timeit np.cumsum(np.arange(1000))
100000 loops, best of 3: 10.1 us per loop
但是,当然,如果这是你唯一使用numpy的地方,则可能不值得依赖它。