一尘不染

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

python

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

阅读 1261

收藏
2020-02-14

共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的地方,则可能不值得依赖它。

2020-02-14