python csv 批量写入

原创

mob64ca12f770a6 2023-12-31 03:39:01 ©著作权

文章标签 CSV 数据 Python 文章分类 Python 后端开发

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

Python CSV 批量写入教程

简介

在数据处理和分析的过程中，经常需要将数据保存到CSV文件中。Python提供了csv模块来支持CSV文件的读写操作。本教程将教会你如何使用Python批量写入CSV文件。

整体流程

首先，让我们来看一下整个流程的步骤：

flowchart TD
    A[准备数据] --> B[打开CSV文件]
    B --> C[写入数据]
    C --> D[关闭CSV文件]

接下来，我们将逐步讲解每个步骤需要做什么，并提供相应的代码和注释。

步骤说明

准备数据

在开始写入CSV文件之前，我们需要准备好要写入的数据。假设我们有一个包含学生信息的列表，每个学生的信息包括姓名、年龄和成绩。让我们创建一个示例数据：

students = [
    {"姓名": "张三", "年龄": 18, "成绩": 90},
    {"姓名": "李四", "年龄": 19, "成绩": 85},
    {"姓名": "王五", "年龄": 20, "成绩": 95}
]

打开CSV文件

接下来，我们需要打开一个CSV文件，以便将数据写入其中。我们可以使用Python内置的open()函数来打开文件。需要注意的是，我们需要指定文件的打开模式为写入模式（"w"）。

import csv

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

写入数据

现在我们已经打开了CSV文件，接下来是将数据写入到文件中。我们可以使用csv.writer对象的writerow()方法来一行一行地写入数据。

# 写入数据
for student in students:
    writer.writerow(student.values())

关闭CSV文件

当我们完成数据写入后，需要关闭CSV文件以确保数据已经正常写入并保存。我们可以使用close()方法来关闭文件。

# 关闭CSV文件
file.close()

完整代码

import csv

students = [
    {"姓名": "张三", "年龄": 18, "成绩": 90},
    {"姓名": "李四", "年龄": 19, "成绩": 85},
    {"姓名": "王五", "年龄": 20, "成绩": 95}
]

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

    # 写入数据
    for student in students:
        writer.writerow(student.values())

# 关闭CSV文件
file.close()

以上代码将会创建一个名为"students.csv"的CSV文件，并将学生信息写入其中。