如何在列表中找到数字的累积和？

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

在Python 2中，您可以定义自己的生成器函数，如下所示：

def accumu(lis):
    total = 0
    for x in lis:
        total += x
        yield total

In [4]: list(accumu([4,6,12]))
Out[4]: [4, 10, 22]

在Python 3.2+中，您可以使用^{}：

In [1]: lis = [4,6,12]

In [2]: from itertools import accumulate

In [3]: list(accumulate(lis))
Out[3]: [4, 10, 22]