嵌套列表中特定项目的总和

Question

我有一个数据文件，例如：

1  123  something else
2  234  something else
3  500  something else
.
. 
.
1  891  something else
2  234  something else
3  567  something else 
.
.
.

我试图最终得到一个文件：

1 1014
2  468
3 1067

也就是说，如果第 1 列中的数字相同，则添加第 2 列（或其他列）中的数字。我相信将列读入嵌套列表并从那里开始是可行的，但我一直在努力解决这个问题。我尝试的另一种方法是使用我感兴趣的条目创建一个新文件：

for next in f.readlines():
    output.write(next[0:1] + "," + next[3:6]+ "\n")
    if not next:
        break

with open(output,"r") as file:
    data_list=[[int(x) for x in line.split(",")] for line in file]

print data_list

返回

[[1, 123], [2, 234], [3, 500], [1, 891], [2, 234], [3, 567]]

我想我可以遍历该列表并比较 data_list[x][0] 并添加匹配的值，但这似乎不是一个优雅的解决方案。 谁能提出一种更优雅的方式来做到这一点？ 特别是，我一直在努力对最终得到的嵌套列表中的特定项目求和。

Answer 1

使用字典来跟踪总和；使用 collections.defaultdict() 可以更容易地从 0 开始键，如果它们以前没有见过：

from collections import defaultdict

sums = defaultdict(int)

with open(filename) as f:
    for line in f:
        col1, col2, rest = line.split(None, 2)
        sums[col1] += int(col2)

这会读取您的初始文件，将空白行拆分 2 次以取出前两列，然后根据第一列对第二列求和：

>>> from collections import defaultdict
>>> sample = '''\
... 1  123  something else
... 2  234  something else
... 3  500  something else
... 1  891  something else
... 2  234  something else
... 3  567  something else 
... '''.splitlines()
>>> sums = defaultdict(int)
>>> for line in sample:
...     col1, col2, rest = line.split(None, 2)
...     sums[col1] += int(col2)
... 
>>> sums
defaultdict(<type 'int'>, {'1': 1014, '3': 1067, '2': 468})