Python-如何找到列表中数字的累计和？

Question

time_interval = [4, 6, 12]

我想对数字求和[4, 4+6, 4+6+12]，以便得到列表t = [4, 10, 22]。

我尝试了以下方法：

for i in time_interval:
    t1 = time_interval[0]
    t2 = time_interval[1] + t1
    t3 = time_interval[2] + t2
    print(t1, t2, t3)

4 10 22
4 10 22
4 10 22

Answer 1

如果你要对像这样的数组做大量的数值工作，我建议使用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的地方，则可能不值得依赖它。