python - (python) 如果条件成立，则在连续行中添加元素

Question

我有一个包含“N”行和“3”列的列表。如果连续行的前两个元素相同，那么我想在第三列中添加元素并返回添加了“第三列”值的一行。

例如

120.638000      -21.541700      0.3  
120.638000      -21.541700      0.8       
121.331001      -21.795500      0.5       
120.688004      -21.587400      0.1        
120.688004      -21.587400      0.5      
120.688004      -21.587400      0.9     
121.525002      -21.504200      0.9    

到

120.638000      -21.541700      1.1  (add third column of row 1 and 2)       
121.331001      -21.795500      0.5       
120.688004      -21.587400      1.5  (sum(0.1,0.5,0.9))       
121.525002      -21.504200      0.9 

对于在 python 中实现这个有什么建议吗？

Answer 1

您可以使用 csvreader 读取您的数据，然后您可以使用 defaultdict 根据 column1,2 中的相同元组对 column3 求和:

from collections import defaultdict
from csv import csvreader

result = defaultdict(float)
with open("<datafile>") as f:
    data = csvreader(f, delimiter='\t')
    for a,b,c in data:
        result[(a,b)] += float(c)

for (a,b),c in result.items():
    print(a, b, c)

这不一定会以相同的顺序出现，因为字典没有排序。