如何使用python中的列表对csv中的列求和

在数据处理过程中，常常需要对 csv 文件进行求和操作，下面就介绍一下如何用 python 中的列表对 csv 文件中的列求和。

假设我们有一个示例的 csv 文件，如下所示（文件名为 example.csv）：

name,age,salary
Tom,25,3000
Jerry,30,4000
Harry,28,5000

我们使用 python 中的 csv 模块来读取该 csv 文件，示例代码如下：

import csv

with open('example.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
        print(row)

运行以上代码，可以看到控制台输出如下：

['name', 'age', 'salary']
['Tom', '25', '3000']
['Jerry', '30', '4000']
['Harry', '28', '5000']

接下来我们可以使用 python 中的列表来处理数据求和。假设我们要对 csv 文件中的第三列（即薪资）求和，代码如下：

import csv

with open('example.csv', 'r') as f:
    reader = csv.reader(f)
    header = next(reader)  # 获取 csv 文件的列标题
    salaries = []  # 用一个列表 salaries 来存储所有员工的薪资
    for row in reader:
        salary = int(row[2])
        salaries.append(salary)  # 将当前员工的薪资添加到 salaries 列表中
    total_salary = sum(salaries)  # 对 salaries 列表进行求和操作
    print("Total salary:", total_salary)

运行以上代码，可以看到控制台输出如下：

Total salary: 12000

以上就是使用 python 中的列表对 csv 文件中的列求和的简单示例。