python csv文件写入列

原创

mob64ca12d12b68 2023-10-07 14:28:25 ©著作权

文章标签 CSV Python 文件写入 文章分类 Python 后端开发

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

Python CSV文件写入列

CSV（Comma Separated Values）是一种常用的文件格式，用于存储和交换表格数据。它是一种纯文本格式，使用逗号来分隔字段。Python提供了许多方法来处理CSV文件，包括读取和写入。

在本文中，我们将重点介绍如何使用Python写入CSV文件的列。我们将使用Python内置的csv模块来完成这个任务。

csv模块简介

csv模块是Python标准库中的一个模块，用于处理CSV文件。它提供了一组方法和类，可以轻松地读取和写入CSV文件。

要使用csv模块，我们首先需要导入它：

import csv

写入CSV文件的列

要写入CSV文件的列，我们首先需要创建一个csv.writer对象。然后，我们可以使用writerow方法将行写入文件。

让我们看一个简单的例子，假设我们有一个包含学生姓名和年龄的列表。我们将把这些信息写入一个名为"students.csv"的CSV文件的两列中。

import csv

students = [
    ['John', 20],
    ['Alice', 19],
    ['Bob', 21]
]

# 创建CSV文件并写入列
with open('students.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['Name', 'Age'])  # 写入列标题
    writer.writerows(students)  # 写入行数据

print("CSV文件写入完成")

在上面的代码中，我们首先创建了一个名为"students.csv"的文件，并以写入模式打开它。我们使用csv.writer创建了一个writer对象。

然后，我们使用writer.writerow方法将列标题写入文件。在本例中，我们使用了一个包含两个元素的列表['Name', 'Age']，它将作为列标题。

接下来，我们使用writer.writerows方法将学生信息列表写入文件。writer.writerows可以接受一个列表的列表作为参数，其中每个内部列表表示一行数据。在本例中，students列表是一个包含学生信息的列表。

最后，我们打印出"CSV文件写入完成"的消息，以表示写入操作已经完成。

甘特图

为了更好地可视化上述代码的执行过程，下面是一个使用mermaid语法标识出的甘特图。

gantt
    dateFormat  YYYY-MM-DD
    title       CSV文件写入列的执行过程

    section 创建CSV文件并写入列
    创建文件           :done, 2022-01-01, 1d
    写入列标题        :done, 2022-01-02, 1d
    写入行数据        :done, 2022-01-03, 2d

以上甘特图清楚地展示了CSV文件写入列的执行过程，包括创建文件、写入列标题和写入行数据。