如何用 Python 将字典写入 CSV

CSV（Comma Separated Values）是一种常见的数据存储格式，可以用于各种应用程序之间的数据交换和共享。Python 中有许多库可以将 Python 数据结构（如列表、字典等）写入 CSV 文件，其中最常用的库之一是 csv 库。

在本篇文章中，我们将介绍如何使用 Python 将字典写入 CSV 文件。

步骤一：准备数据

首先，我们需要准备要写入 CSV 文件的字典数据。假设我们有以下字典数据：

data = {'name': ['张三', '李四', '王五'], 'age': [20, 25, 30], 'gender': ['男', '女', '男']}

这个字典包含三个键值对，键分别为 name、age、gender，值分别为包含姓名、年龄和性别的列表。

步骤二：创建 CSV 文件

要将字典写入 CSV 文件，我们需要先创建 CSV 文件。可以使用内置的 open 函数打开文件，并使用 csv.writer 函数创建一个写入器对象。下面的代码示例创建名为 data.csv 的 CSV 文件：

import csv
 
with open('data.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    # 写入表头
    writer.writerow(data.keys())

参数 mode 设置为 ‘w’ 表示以写入模式打开文件，如果文件不存在则创建它；参数 newline 设置为空表示不使用额外的行结束符。使用 csv.writer 函数创建的写入器对象可以将数据写入 CSV 文件。

在上述代码中，writer.writerow(data.keys()) 将字典的键写入 CSV 文件，即将“name”、“age”和“gender”写入文件的第一行，作为表头。

步骤三：写入数据

现在我们可以使用 csv.writer 的 writerow 方法将数据写入 CSV 文件。下面的代码示例将 data 字典的值写入 CSV 文件：

import csv
 
with open('data.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    # 写入表头
    writer.writerow(data.keys())
    # 写入数据
    writer.writerows(zip(*data.values()))

代码中使用了 zip 函数将字典的值打包成元组，再使用 * 号展开元组，将三个列表转换成三个独立的参数。这样就可以将每个同名键的值一一对应起来，然后使用 writer.writerows 方法将这些对应值写入文件。

代码执行完毕后，CSV 文件中的数据如下：

name,age,gender
张三,20,男
李四,25,女
王五,30,男

注意事项

1. 在写入 CSV 文件之前，需要先准备要写入的数据。可以使用列表、元组、字典等数据类型。
2. 在使用 csv.writer 函数创建写入器对象时，需要指定要写入的 CSV 文件的名称、文件打开模式和是否添加额外的行结束符。
3. 在使用 writer.writerows 方法写入数据时，需要将数据打包成一行一行的元组，并将多个元组组合成一个列表，才能将它们写入 CSV 文件。
4. 如果字典中有空值或缺失值，编写代码时需要进行适当的数据处理，避免写入 CSV 文件时出现错误。

火焰兔

海报 链接