Python - 将列表保存为 CSV

当我们需要将一个列表中的数据保存到 CSV 文件中时，Python 中的 csv 模块提供了很好的支持。本文将介绍如何使用 csv 模块将列表保存为 CSV 文件。

将列表保存为 CSV 文件

要将列表保存为 CSV 文件，我们需要导入 csv 模块，并使用 csv.writer() 来创建一个 csv.writer 对象。然后，我们可以使用 csv.writer 对象的 writerow() 方法，将列表中的每一行数据写入到 CSV 文件中。

import csv

# 定义一个列表
data = [
    ['姓名', '年龄', '爱好'],
    ['张三', 20, '篮球'],
    ['李四', 25, '游泳'],
    ['王五', 30, '跑步']
]

# 打开 CSV 文件，并写入数据
with open('data.csv', 'w', encoding='utf-8', newline='') as csv_file:
    writer = csv.writer(csv_file)
    for row in data:
        writer.writerow(row)

在上面的代码中，我们先定义一个列表 data，其中包含了表格的所有数据。然后，我们使用 csv.writer() 创建一个 csv.writer 对象，并将列表中的每一行数据写入到 CSV 文件 data.csv 中。

读取 CSV 文件

如果我们要读取 CSV 文件中的数据，只需要使用 csv.reader() 方法即可，它可以将 CSV 文件中的每一行数据转换成一个列表。

import csv

# 读取 CSV 文件中的数据
with open('data.csv', 'r', encoding='utf-8') as csv_file:
    reader = csv.reader(csv_file)
    for row in reader:
        print(row)

在上面的代码中，我们使用 csv.reader() 方法读取 CSV 文件 data.csv 中的数据，并将每一行数据转换成一个列表。最后，我们使用 print() 方法将每一行数据打印出来。

结语

通过上面的介绍，相信你已经了解了如何使用 Python 将列表保存为 CSV 文件，并读取 CSV 文件中的数据了。如有疑问或建议，欢迎在评论区留言。