隔离列表中的两个索引并将其中一个索引与另一个索引进行求和？

Question

我有与内容

[100,123,111] 
[010,789,222] 
[001,123,222] 
[230,789,234] 

我想要的号码的索引2，其在索引1对于上述示例共享相同的数目的总和，期望输出是

[333,456] 

文本文件

因为111 + 222 = 333（在索引1共享数：123），并且因为222 + 234 = 456。

Answer 1

确定，这里是与ast.literal_eval和collections.OrderedDict的方法。

让lists.txt与

[10]
[010789222]
[001123222]
[230789234]

>>> from ast import literal_eval 
>>> from collections import OrderedDict 
>>> with open('lists.txt') as f: 
...  d = OrderedDict() 
...  for line in f: 
...   lst = literal_eval(line) 
...   d[lst[1]] = d.get(lst[1], 0) + lst[2] 
... 
>>> d.values() 
[333, 456] 

Answer 2

我使用的内容文件一个defaultdict缩短添加元素到字典的代码。我只是保持索引1作为密钥的运行总数，然后打印出值

from collections import defaultdict 

l = [[100,123,111], 
    [010,789,222], 
    [001,123,222], 
    [230,789,234]] 

d = defaultdict(int) 

for i in l: 
    d[i[1]] = d[i[1]]+i[2] 

print d.values() 

[333, 456]