隔離列表中的兩個索引並將其中一個索引與另一個索引進行求和？

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]