查找CSV文件的总和

Question

编写一个名为“ csv_sum”的函数，该函数以字符串作为参数，表示格式为“ 、、、”的5列CSV文件的名称，并从中返回第五列中所有值的总和。输入文件

import csv
def csv_sum(st):
    val=0
    with open(st)as a:
        reader=csv.reader(a)
        for line in reader:
            for b in line:
                val=val+int(line[4])
    return val

我不完全了解如何使用CSV文件，因此我在上面的尝试是解决它的尝试。但是我在输入['animal.csv']

时遇到了错误

示例...

animal.csv

9,16,136,-24,3

3,13,170,-21,4

返回：35

预期：7

Answer 1

只需要第5列时，就不必要遍历一行中的每一列。删除for b in line语句，您的代码应该可以运行：

def csv_sum(st):
    val=0
    with open(st)as a:
        reader=csv.reader(a)
        for line in reader:
            val=val+int(line[4])
    return val