我正在尝试在csv文件中创建一个列的总和.该文件看起来像:
- Date Value
- 2012-11-20 12
- 2012-11-21 10
- 2012-11-22 3
这可以在数百行的范围内.我需要在终端上打印出的总值(在这种情况下它将是25).到目前为止,我有一些代码,但它导致的数字要小得多.在对其进行故障排除时,我进行了总和的打印,并意识到它实际上打破了每列中的数字并且总和为1 2 1 0 3,而不是总和12 10 3,这显然等于更小的总数.这是我的代码,如果有人可以提出建议会很棒!
- with open("file.csv")) as fin:
- headerline = fin.next()
- total = 0
- for row in csv.reader(fin):
- print col # for troubleshooting
- for col in row[1]:
- total += int(col)
- print total
解决方法
csv模块逐个循环遍历您的行,因此无需在列上循环.只是总和int(行[1]):
- with open("file.csv") as fin:
- headerline = fin.next()
- total = 0
- for row in csv.reader(fin):
- total += int(row[1])
- print total
- with open("file.csv") as fin:
- fin.next()
- total = sum(int(r[1]) for r in csv.reader(fin))
请注意,在Python中,字符串也是序列,所以当您在第[1]行中执行col时:您正在遍历row [1]的各个字符;所以你的第一行是1和2:
- >>> for c in '123':
- ... print repr(c)
- ...
- '1'
- '2'
- '3'