Python列表积累

时间:2014-06-18 19:03:34

标签: python list accumulate

因此,以下代码读取两个文本文件,这两个文件在25行中包含25个整数,并将它们放在两个相应的列表中:

num_DIR = '/path/to/num.txt'
den_DIR = '/path/to/den.txt'

def makeList(DIR):
    list = []
    for line in open(DIR).readlines():
            list.append(line.strip())
    return list

num_list = makeList(num_DIR)
den_list = makeList(den_DIR)

对外输出:

num_list = ['76539', '100441', '108637', '108874', '103580', '91869', '78458', '61955', '46100', '32701', '21111', '13577', '7747', '4455', '2309', '1192', '554', '264', '134', '63', '28', '15', '12', '7', '5']

den_list = ['621266', '496647', '436229', '394595', '353249', '305882', '253983', '199455', '147380', '102872', '67255', '41934', '24506', '13778', '7179', '3646', '1778', '816', '436', '217', '114', '74', '49', '44', '26']

我如何将两个列表中的每个值都添加到其后的所有值的总和中,如在累积列表中一样?

3 个答案:

答案 0 :(得分:2)

如果表现不重要,您还可以使用sum和列表理解:

>>> num_list=[1,2,3]
>>> [sum(num_list[:y]) for y in range(1, len(num_list) + 1)]
[1, 3, 6]

答案 1 :(得分:2)

如果您运行的是python 3.2或更高版本,则itertools.accumulate中包含电池。

如果您之前运行的是某个版本,那么文档中提供的pure-python等效工作等效(尽管不是C速度):

def accumulate(iterable, func=operator.add):
    'Return running totals'
    # accumulate([1,2,3,4,5]) --> 1 3 6 10 15
    # accumulate([1,2,3,4,5], operator.mul) --> 1 2 6 24 120
    it = iter(iterable)
    total = next(it)
    yield total
    for element in it:
        total = func(total, element)
        yield total

演示:

from itertools import accumulate

list(accumulate(map(int,num_list)))
Out[3]: 
[76539,
 176980,
 285617,
 394491,
 498071,
 589940,
 668398,
 730353,
 776453,
 809154,
 830265,
 843842,
 #snip...,
 860627
]

请注意,您需要将所有字符串转换为int s,就像我使用map一样。

答案 2 :(得分:2)

您似乎希望在每个元素之后将所有值的总和添加到列表的末尾

summed=[]
den_list = map(int,den_list)
for i,j in enumerate(den_list[:-1]):
    summed += j,sum(den_list[i+1:])

如果您不想将原始值保留在列表中

summed=[sum(num_list[i+1:])for i,j in enumerate(num_list[:-1])]