我有一个大名单，其中的一个片段是这样的：
power = [
['1234-43211', [5, 6, -4, 11, 22]],
['1234-783411', [43, -5, 0, 0, -1]],
['1234-537611', [3, 0, -5, -6, 0]],
['1567-345411', [4, 6, 8, 3, 3]],
['1567-998711', [1, 2, 1, -4, 5]]
]
字符串中的第一个数字是重要的，而且在我希望我分开的附加一个。 即我只希望每个站内会逐渐增加值（和返回每个单数累加），从来没有从两个不同的值相加。
我的目标是要遍历这个列表，并添加累计整型值一站，回报每此外，在列表中检测到下一站的时候再重新开始。
期望的结果：
new = [
[48, 1, -4, 11, -21],
[ 51, 1, -9, 5, -21], '### End of '1234' ### '
[5, 8, 9, -1, 8], '### End of 1567 ###'
] or something similar to this
我曾尝试以下：
for i in range(len(power)-1):
front_num_1 = power[i][0].split('-')[0]
front_num_2 = power[i+1][0].split('-')[0]
station = '%s' % (front_num_1)
j = power[i][1]
k = power[i+1][1]
if front_num_1 == front_num_2:
print [k + j for k, j in zip(j, k)]
elif front_num_1 != front_num_2:
print '#####################################
else:
print 'END'
然而，这除了不具有累积性，因此没有用。
from itertools import groupby, islice
def accumulate(iterable): # in py 3 use itertools.accumulate
''' Simplified version of accumulate from python 3'''
it = iter(iterable)
total = next(it)
yield total
for element in it:
total += element
yield total
power = [
['1234-4321-1', [5, 6, -4, 11, 22]],
['1234-7834-1', [43, -5, 0, 0, -1]],
['1234-5376-1', [3, 0, -5, -6, 0]],
['1567-3454-1', [4, 6, 8, 3, 3]],
['1567-9987-1-', [1, 2, 1, -4, 5]]
]
groups = ((k, (nums for station, nums in g))
for k, g in
groupby(power, lambda x: x[0].partition('-')[0]))
new = [(station, zip(*(islice(accumulate(col), 1, None) for col in zip(*nums))))
for station, nums in groups]
print new
print dict(new) # or as a dictionary which is unordered
产量
[('1234', [(48, 1, -4, 11, 21), (51, 1, -9, 5, 21)]), ('1567', [(5, 8, 9, -1, 8)])]
{'1234': [(48, 1, -4, 11, 21), (51, 1, -9, 5, 21)], '1567': [(5, 8, 9, -1, 8)]}
这是如何工作：
首先，列表是基于使用站分组itertools.groupby 。
例如。
nums = [[5, 6, -4, 11, 22],
[43, -5, 0, 0, -1],
[3, 0, -5, -6, 0]]
是第一组。 正如你可以看到它是矩阵的形式。
zip(*nums)调换使用参数拆包的矩阵。 它要求
zip([5, 6, -4, 11, 22], [43, -5, 0, 0, -1], [3, 0, -5, -6, 0])
它创建列表：
cols = [(5, 43, 3), (6, -5, 0), (-4, 0, -5), (11, 0, -6), (22, -1, 0)]
然后积聚调用的每个列，下面是什么样子：
>>> [list(accumulate(col)) for col in cols]
[[5, 48, 51], [6, 1, 1], [-4, -4, -9], [11, 11, 5], [22, 21, 21]]
正如你在这里每个列表中看到的第一个元素，因而不需要islice用于取从索引元件1直到然后结束（ None ）。 这里是什么样子：
>>> [list(islice(accumulate(col), 1, None)) for col in cols]
[[48, 51], [1, 1], [-4, -9], [11, 5], [21, 21]]
现在，我们只需要转这个了。
>>> zip(*(islice(accumulate(col), 1, None) for col in cols))
[(48, 1, -4, 11, 21), (51, 1, -9, 5, 21)]
如果你打破了你的问题成小块这将有助于。 我似乎明白要1）基于一些标准划分您的列表，然后2）取各子表（考虑每个元素的矢量）的累计总和。
例如：
stationList = [
['1234-4321-1', [5, 6, -4, 11, 22]],
['1234-7834-1', [43, -5, 0, 0, -1]],
['1234-5376-1', [3, 0, -5, -6, 0]],
['1567-3454-1', [4, 6, 8, 3, 3]],
['1567-9987-1-', [1, 2, 1, -4, 5]]
]
变为：
{'1234-4321-1': [
<5, 6, -4, 11, 22>,
<5, 6, -4, 11, 22> + <43, -5, 0, 0, -1>,
<5, 6, -4, 11, 22> + <43, -5, 0, 0, -1> + <3, 0, -5, -6, 0>
],
'1567-3454-1': [
<4, 6, 8, 3, 3>,
<4, 6, 8, 3, 3> + <1, 2, 1, -4, 5>
]
}
（其中，我使用<...>来表示一个假想Vector对象，或仅处理该列表作为向量）。
解
from itertools import *
1），为了将基于一些标准列表，使用itertools.groupby： 文档在这里 。 或者写一个发生器功能。
getStation = lambda x: x[0].split('-')[0]
def groupby_station(inputList):
return groupby(inputList, key=getStation)
2）的累加值可以被写入作为发电机的功能。 您可以使用numpy ，或者只是把它写自己。
def listAdd(*lists):
"""
listAdd([1,2,3], [10,20,30]) -> [11,22,33]
listAdd([1,2,3], []) -> [1,2,3]
"""
return [sum(xs) for xs in zip_longest(*lists, fillvalue=0)]
def cumSum(lists):
"""
cumSum([1,2],[10,20],[100,200]) -> ([1,2],[11,22],[111,222])
"""
total = []
for list in lists:
total = listAdd(total, list)
yield total
现在，只要将二者结合起来：
{key:cumSum(*lists) for key,lists in groupby_station(inputList)}
请注意，我累计总和的定义是从你的略有不同; 您可以修改cumSum功能，以配合您的定义。
