如何使用 python 函数读取和写入 CSV 文件

CSV（Comma Separated Values）是一种常见的数据交换格式，通常用逗号作为字段之间的分隔符。在数据分析和处理中，读取和写入 CSV 文件是非常常见的任务。Python 提供了很多内置模块和第三方库来处理 CSV 文件，本文将介绍如何使用 Python 函数读取和写入 CSV 文件。

1. 读取 CSV 文件

使用 Python 内置的 csv 模块可以轻松地读取 CSV 文件。 csv 模块提供了一个 DictReader 类，可以将 CSV 文件中的每一行转换成一个 Python 字典，其中每个键都是 CSV 文件的表头，每个值都是对应的表格中的值。以下是使用 Python 读取 CSV 文件的示例：

import csv

with open("data.csv", "r") as f:
    reader = csv.DictReader(f)
    for row in reader:
        print(row["name"], row["age"], row["gender"])

在上面的示例中，我们打开一个名为 data.csv 的 CSV 文件，并使用 DictReader 类读取文件的每一行。然后，我们可以使用 Python 的 for 循环遍历每一行，并将每一行转换成一个字典。最后，我们可以使用字典中的键来访问每个值，例如 row["name"]、row["age"] 和 row["gender"]。

2. 写入 CSV 文件

使用 Python 也很容易将数据写入 CSV 文件。 csv 模块提供了一个 DictWriter 类，可以将 Python 字典转换为 CSV 文件的每一行。以下是使用 Python 将数据写入 CSV 文件的示例：

import csv

data = [
    {"name": "John", "age": 25, "gender": "Male"},
    {"name": "Sarah", "age": 30, "gender": "Female"},
    {"name": "Tom", "age": 35, "gender": "Male"}
]

with open("data.csv", "w", newline="") as f:
    fieldnames = ["name", "age", "gender"]
    writer = csv.DictWriter(f, fieldnames=fieldnames)
    writer.writeheader()
    for row in data:
        writer.writerow(row)

在上面的示例中，我们先定义了一个包含三个 Python 字典的列表，每个字典包含一个人的姓名、年龄和性别。然后，我们打开一个名为 data.csv 的 CSV 文件并使用 DictWriter 类来写入数据。首先，我们使用 writer.writeheader() 函数写入表头，即每一列的名称。然后，我们使用 for 循环遍历数据列表中的每个 Python 字典，并使用 writer.writerow(row) 函数写入每一行数据。

总结

CSV 文件在数据分析和处理中非常常见，Python 提供了 csv 模块来处理 CSV 文件。使用 csv 模块可以轻松地读取和写入 CSV 文件，并将 CSV 文件转换为 Python 字典。以上是使用 Python 函数读取和写入 CSV 文件的基本方法，可以根据具体需求进行修改和扩展。