python将csv的每一行求和_如何在python中更有效地对csv文件中的列进行求和

python defaultdict

所以，你会发现很多有用的例子，下面是我的答案import csv

from collections import defaultdict

# slurp the data

data = list(csv.reader(open('points.csv')))

# massage the data

for i, row in enumerate(data[1:],1):

data[i] = [int(elt) if elt.isdigit() else elt for elt in row]

points = {} # an empty dictionary

for i, country in enumerate(data[0][2:],2):

# for each country, a couple country:defaultdict is put in points

points[country] = defaultdict(int)

for row in data[1:]:

opponent = row[1]

points[country][opponent] += row[i]

# here you can post-process points as you like,

# I'll simply print out the stuff

for country in points:

for opponent in points[country]:

print country, "vs", opponent, "scored",

print points[country][opponent], "points."

数据的示例输出是

^{pr2}$

编辑

如果您反对defaultdict，那么可以使用普通dict的.get方法，该方法允许您在{}对未初始化的情况下返回可选的默认值points[country] = {} # a standard empty dict

for row in data[1:]:

opponent = row[1]

points[country][opponent] = points[country].get(opponent,0) + row[i]

如您所见，它有点笨拙，但仍然可以管理。在