python csv逐行写入

原创

mob649e815a6b81 2024-08-27 04:44:29 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e815a6b81的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python 中 CSV 逐行写入的教程

在现代数据处理中，CSV（逗号分隔值）文件被广泛用于存储和交换数据。如果你是一名新手开发者，想要学习如何使用 Python 逐行写入 CSV 文件，本文将为你提供一个详细的指南。

整体流程

在开始编写代码之前，我们首先明确整个过程的步骤。我们可以用一个表格来描述这个流程：

步骤	描述
1	导入 `csv` 模块
2	打开一个 CSV 文件（创建或追加模式）
3	创建一个 CSV 写入对象
4	逐行写入数据
5	关闭文件

每一步的详细实现

现在，我们将逐步实现上述流程，并解释每一步需要使用的代码。

步骤 1：导入 `csv` 模块

首先，我们需要导入 Python 中的 csv 模块，该模块提供了用于读写 CSV 文件的功能。

import csv  # 导入 CSV 模块

步骤 2：打开 CSV 文件

接下来，我们需要打开一个 CSV 文件。可以选择创建一个新文件，或者以追加的方式打开一个已存在的文件。

with open('data.csv', mode='a', newline='') as file:  # 以追加模式打开文件
    # 接下来的代码将在该上下文中运行

mode='a' 表示以追加模式打开文件；如果文件不存在，将会创建新文件。
newline='' 参数用于避免在 Windows 系统上添加多余的换行符。

步骤 3：创建 CSV 写入对象

在成功打开文件后，我们需要创建一个 CSV 写入对象，以便我们可以方便地将数据写入 CSV 文件。

    writer = csv.writer(file)  # 创建 CSV 写入对象

步骤 4：逐行写入数据

现在，我们可以逐行写入数据了。你可以在一个循环中将数据写入列表。

    # 示例数据，可以替换为你自己的行数据
    data = [
        ["Name", "Age", "City"],  # 表头
        ["Alice", 30, "New York"],
        ["Bob", 25, "Los Angeles"],
        ["Charlie", 28, "Chicago"]
    ]

    for row in data:  # 遍历每一行数据
        writer.writerow(row)  # 将当前行写入 CSV 文件

步骤 5：关闭文件

由于我们使用 with 语句打开文件，文件将在离开上下文后自动关闭。因此这里不需要手动关闭文件。

完整代码示例

以下是将上述所有步骤整合起来的完整代码：

import csv  # 导入 CSV 模块

# 打开 CSV 文件，以追加模式
with open('data.csv', mode='a', newline='') as file:
    writer = csv.writer(file)  # 创建 CSV 写入对象

    # 示例数据
    data = [
        ["Name", "Age", "City"],  # 表头
        ["Alice", 30, "New York"],
        ["Bob", 25, "Los Angeles"],
        ["Charlie", 28, "Chicago"]
    ]

    # 遍历数据并逐行写入
    for row in data:  
        writer.writerow(row)  # 将当前行写入 CSV 文件

旅行图示例

接下来，我们用 Mermaid 语法展示一个简单的旅行图，表示这个过程。

journey
    title Python CSV 逐行写入流程
    section 初始化
      导入 CSV 模块: 5: 用户
    section 打开文件
      打开 CSV 文件: 4: 用户
    section 写入数据
      创建 CSV 写入对象: 4: 用户
      逐行写入数据: 4: 用户
    section 完成
      关闭文件: 5: 用户