如何使用Python的CSV函数库读写CSV文件

Python提供了csv模块来读写CSV文件。csv模块包含了一些函数和类，可以帮助我们方便地读取和写入CSV文件。

1. 导入csv模块

首先需要导入csv模块，在代码的开头添加以下语句：

import csv

2. 读取CSV文件

要读取CSV文件，可以使用csv模块中的DictReader类。该类会将每一行数据转换为一个字典，使得我们可以通过键值对的方式来访问每一列的数据。以下是读取CSV文件的代码示例：

with open('data.csv', 'r') as file:

    csv_reader = csv.DictReader(file)

    for row in csv_reader:

        print(row['name'], row['age'])

在这个示例中，我们打开名为data.csv的文件，并创建了一个DictReader对象来读取文件内容。通过遍历DictReader对象，我们可以访问每一行数据，并使用键值对来获取每列的数据。

3. 写入CSV文件

要写入CSV文件，可以使用csv模块中的DictWriter类。该类可以将字典数据写入CSV文件。以下是写入CSV文件的代码示例：

data = [{'name': 'Alice', 'age': '25'}, {'name': 'Bob', 'age': '30'}]

with open('data.csv', 'w', newline='') as file:

    fieldnames = ['name', 'age']

    csv_writer = csv.DictWriter(file, fieldnames=fieldnames)

    csv_writer.writeheader()

    csv_writer.writerows(data)

在这个示例中，我们首先将要写入的数据存储在一个字典列表data中。然后，我们打开一个名为data.csv的文件，并创建了一个DictWriter对象来写入文件内容。通过调用writeheader方法，我们可以写入CSV文件的表头。最后，通过调用writerows方法，我们可以写入数据。

需要注意的是，open函数的第三个参数newline=''是为了防止写入文件时出现空行问题。

通过以上的步骤，我们可以方便地使用Python的csv模块来读取和写入CSV文件。