Python 字典写入 CSV

CSV（Comma Separated Values）是一种常用的文件格式，通常用于存储表格数据。Python 提供了许多库来操作 CSV 文件，其中最常用的是 csv 模块。

在实际工作中，我们常常需要将 Python 字典中的数据写入到 CSV 文件中，以进行数据的备份和分享。本文将介绍如何使用 Python 将字典数据写入 CSV 文件中，供大家参考。

创建字典数据

在本文中，我们将使用以下字典数据作为示例：

data = [
    {'name': 'Alice', 'age': 25, 'gender': 'female'},
    {'name': 'Bob', 'age': 30, 'gender': 'male'},
    {'name': 'Charlie', 'age': 35, 'gender': 'male'}
]

以上字典数据包含了三个人的姓名、年龄和性别。接下来，我们将使用 Python 将该字典数据写入 CSV 文件中。

创建 CSV 文件

在开始写入数据之前，我们需要创建一个 CSV 文件，并将数据写入该文件中。以下是创建文件的代码：

import csv

filename = 'data.csv'
with open(filename, 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['name', 'age', 'gender'])

在上面的代码中，我们使用 csv.writer 创建了一个写入文件的对象，并使用 writerow 方法写入了表头行。在上面的代码中，我们指定了文件名（'data.csv'）和写入模式（'w'）。同时，我们使用 newline='' 参数来避免写入空行。

写入字典数据

接下来，我们将使用 csv.writer 对象将字典数据写入文件中。以下是写入数据的代码：

import csv

data = [
    {'name': 'Alice', 'age': 25, 'gender': 'female'},
    {'name': 'Bob', 'age': 30, 'gender': 'male'},
    {'name': 'Charlie', 'age': 35, 'gender': 'male'}
]

filename = 'data.csv'
with open(filename, 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['name', 'age', 'gender'])
    for row in data:
        writer.writerow(row.values())

在上面的代码中，我们使用 for 循环遍历字典数据，将每个字典中的值写入到 CSV 文件中。我们使用了 row.values() 方法来获取每个字典中的值。

读取 CSV 文件

在将数据写入 CSV 文件后，我们可以使用以下代码来读取文件中的数据：

import csv

filename = 'data.csv'
with open(filename, newline='') as file:
    reader = csv.DictReader(file)
    for row in reader:
        print(row)

在上面的代码中，我们使用 csv.DictReader 对象读取 CSV 文件，并将其转换为一个字典对象。我们使用 for 循环遍历该对象，并打印出每行数据。输出结果如下：

{'name': 'Alice', 'age': '25', 'gender': 'female'}
{'name': 'Bob', 'age': '30', 'gender': 'male'}
{'name': 'Charlie', 'age': '35', 'gender': 'male'}

在上述输出结果中，我们可以看到 CSV 文件中的每行数据被转换为了一个 Python 字典对象。

结论

在本文中，我们介绍了如何使用 Python 将字典数据写入到 CSV 文件中，并介绍了如何读取 CSV 文件并将其转换为 Python 字典对象。CSV 文件是一种常用的数据格式，对于数据的备份和分享都有很大的帮助。希望本文对大家有所帮助。