Python 操作 CSV 文件实例详解

什么是 CSV 文件？

CSV 是指逗号分隔值（Comma-Separated Values），是一种常见的电子表格文件格式，通常以 .csv 作为文件后缀。CSV 文件由以逗号分隔的多行数据组成，常用来存储数据以供程序读取。

Python 操作 CSV 文件

Python 标准库中提供了 csv 模块，该模块可以帮助我们读取、写入和操作 CSV 文件。下面我们将介绍如何使用 csv 模块来读取和写入 CSV 文件。

读取 CSV 文件

下面是一个简单的 CSV 文件 example.csv，它的内容包含了两行数据：

Name,Age,Country
Alice,20,USA
Bob,30,Canada

我们可以使用 csv 模块中的 reader 方法来读取该文件：

import csv

with open('example.csv', newline='') as csvfile:
    reader = csv.reader(csvfile, delimiter=',', quotechar='"')
    for row in reader:
        print(', '.join(row))

输出如下：

Name, Age, Country
Alice, 20, USA
Bob, 30, Canada

上述代码中，我们使用了 with open 语句来打开 example.csv 文件，csv.reader 方法用于解析 CSV 文件内容，delimiter 参数用于指定 CSV 文件中的数据分隔符，quotechar 参数用于指定 CSV 文件中的数据包含符。

上述代码中，我们将每一行 CSV 数据合并成一个字符串，并用逗号进行分隔。

写入 CSV 文件

除了读取 CSV 文件外，csv 模块还可以用于将数据写入到 CSV 文件中。下面是一个简单的示例，演示如何将数据写入到 CSV 文件中：

import csv

data = [
    ['Name', 'Age', 'Country'],
    ['Alice', '20', 'USA'],
    ['Bob', '30', 'Canada']
]

with open('example.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile, delimiter=',', quotechar='"')
    for row in data:
        writer.writerow(row)

上述代码中，我们首先定义了一个列表 data，其中包含了三个列表，每个列表包含了三个数据项，它们分别表示名称、年龄和所属国家。然后，我们使用 with open 语句来创建一个 CSV 文件，csv.writer 函数用于将数据写入 CSV 文件中，delimiter 和 quotechar 参数同样用于指定 CSV 文件中的数据分隔符。

运行上述代码后，将会在当前目录下创建一个名为 example.csv 的文件，并将数据项写入到该文件中。

总结

本文中，我们介绍了如何使用 Python 的 csv 模块来读取和写入 CSV 文件，并提供了相应的示例代码。需要注意的是，在读写 CSV 文件时，我们需要正确指定 CSV 文件中的数据分隔符和数据包含符，才能够正确地解析和写入 CSV 数据。