Python循环写CSV

CSV（Comma-Separated Values）是一种常见的文件格式，用于存储表格数据。在Python中，我们可以使用循环来写入CSV文件。本文将介绍如何使用Python循环写入CSV文件，并附带代码示例。

准备工作

在开始编写代码之前，我们需要安装csv模块，如果你的Python环境中没有安装该模块，可以使用以下命令进行安装：

pip install csv

代码示例

下面是一个简单的示例代码，演示了如何使用循环写入CSV文件。假设我们有一个包含学生信息的列表，每个学生的信息包括姓名、年龄和成绩。我们要将这些信息写入一个CSV文件中。

import csv

students = [
    {"name": "Alice", "age": 18, "score": 90},
    {"name": "Bob", "age": 19, "score": 85},
    {"name": "Charlie", "age": 20, "score": 95}
]

# 定义CSV文件的列名
fieldnames = ["Name", "Age", "Score"]

# 打开CSV文件
with open("students.csv", "w", newline="") as csvfile:
    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)

    # 写入列名
    writer.writeheader()

    # 循环写入每个学生的信息
    for student in students:
        writer.writerow(student)

在这个示例中，我们首先定义了一个包含学生信息的列表students。然后，我们定义了CSV文件的列名fieldnames，即表头信息。接着，我们使用open函数打开一个名为students.csv的文件，并将其赋值给变量csvfile。通过csv.DictWriter函数和fieldnames参数，我们创建了一个writer对象，用于写入CSV文件。

使用writer.writeheader()方法，我们写入了CSV文件的列名。然后，我们使用循环遍历students列表中的每个学生，使用writerow方法将每个学生的信息写入CSV文件。

最后，我们使用with语句来自动关闭文件，以确保资源的正确释放。

序列图

下面是使用Mermaid语法绘制的序列图，展示了代码中的流程：

sequenceDiagram
    participant A as Code
    participant B as CSV File
    
    A ->> B: 打开CSV文件
    A ->> B: 写入列名
    loop 每个学生的信息
        A ->> B: 写入学生信息
    end
    A ->> B: 关闭CSV文件

总结

本文介绍了如何使用Python循环写入CSV文件。我们首先准备工作，安装了csv模块。然后，我们使用示例代码演示了如何使用循环和csv模块将学生信息写入CSV文件。最后，我们使用Mermaid语法绘制了序列图，展示了代码的执行流程。

通过使用循环写入CSV文件，我们可以方便地将大量数据写入CSV文件，实现数据的持久化存储和交换。希望本文对你有帮助！