python - 如何汇总CSV列？

Question

我想使用下面的 python 脚本对 CSV 文件中的一列进行总计。但是，我收到以下错误。

invalid literal for int() with base 10: '7.3'

我的脚本有什么问题？

import csv

cr = csv.reader(open("companylist.csv","rb"))
cr.next() # to skip the header 

total = 0
for row in cr:  
   total += int(row[2])
   # possibly do other things with data/rows 

print total

Answer 1

您可以通过 Generator Expression 一次完成，而不是创建 total 变量并递增它。 .

import csv

with open("book1.csv", "rb") as csv_file:
    reader = csv.DictReader(csv_file)
    total = sum(float(row["column_name_here"]) for row in reader)
print(total)