Python将list写入CSV

在数据分析和处理中，我们经常需要将数据存储到不同的格式中，如CSV（逗号分隔值）格式。CSV是一种常见的文本格式，用于存储表格数据。Python中有许多库可以用于处理CSV文件，其中最常用的是csv库。本文将介绍如何使用Python的csv库将列表（list）写入CSV文件。

准备工作

首先，我们需要安装Python的csv库。如果你使用的是Python的标准安装包，那么csv库已经包含在内。如果你使用的是其他Python发行版，或者没有安装该库，你可以使用以下命令来安装：

pip install csv

创建数据列表

在开始之前，我们需要创建一个包含数据的列表。我们以一个简单的例子开始，假设我们有一个包含学生姓名和年龄的列表。我们可以使用以下代码创建一个名为students的列表：

students = [["Alice", 21],
            ["Bob", 22],
            ["Charlie", 20],
            ["David", 19]]

在这个例子中，我们创建了一个包含4个学生的列表。每个学生是一个包含姓名和年龄的子列表。

将列表写入CSV文件

有了数据列表后，我们可以使用csv库将其写入CSV文件。下面是一个示例代码，展示了如何将students列表写入名为students.csv的CSV文件中：

import csv

filename = "students.csv"

with open(filename, 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerows(students)

让我们逐行分析这个代码：

  1. 首先，我们导入了csv库。
  2. 接着，我们指定了要写入的CSV文件的名称为students.csv。
  3. 然后，我们使用open()函数打开文件，并将其赋值给一个变量csvfile。
  4. 使用csv.writer()函数创建一个writer对象，该对象用于将数据写入CSV文件。
  5. 最后，我们使用writer.writerows()方法将整个students列表写入CSV文件。

这样，我们就成功将数据列表写入CSV文件中了。

运行结果

当我们运行上面的代码后，将在代码所在的目录下创建一个名为students.csv的CSV文件。打开该文件，我们可以看到以下内容：

Alice,21
Bob,22
Charlie,20
David,19

每个学生的姓名和年龄被逗号分隔开来，并且每个学生在文件中占据一行。

总结

本文介绍了如何使用Python的csv库将列表写入CSV文件。首先，我们准备了一个包含学生姓名和年龄的列表。然后，我们使用csv库的writer对象将数据写入CSV文件中。最后，我们展示了运行结果。希望本文对你在Python中处理CSV文件时有所帮助。

参考文献

  • [Python CSV Documentation](