使用 Python 处理 CSV 文件

使用 Python 处理 CSV 文件

简介

这份文档将讲解如何使用 Python 编程语言来处理 CSV 文件。CSV 文件是一种常用的数据存储格式，它将数据以逗号分隔的形式存储在文本文件中。Python 拥有强大的库可以轻松地读取、写入和操作 CSV 文件。

依赖库

为了处理 CSV 文件，你需要使用 Python 的 csv 库。该库提供了一系列函数，可以帮助你读取、写入和操作 CSV 文件。你可以通过以下代码导入 csv 库：

import csv

读取 CSV 文件

使用 csv 库的 reader 对象可以读取 CSV 文件的内容。reader 对象会将文件内容解析成一个二维列表，其中每一行表示 CSV 文件中的一个数据行，每一列表示数据行中的一个值。

with open('data.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row)

这段代码将读取 data.csv 文件的内容，并打印每一行数据。

写入 CSV 文件

可以使用 csv 库的 writer 对象将数据写入 CSV 文件。writer 对象可以将二维列表或其他可迭代对象写入 CSV 文件。

with open('output.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['姓名', '年龄', '城市'])
    writer.writerows([['张三', 25, '北京'], ['李四', 30, '上海']])

这段代码将创建一个名为 output.csv 的 CSV 文件，并在文件中写入三行数据。

操作 CSV 文件

除了读取和写入之外，还可以使用 csv 库对 CSV 文件进行其他操作，例如：

• 跳过头部行: 使用 next(reader) 方法可以跳过 CSV 文件中的第一行，通常这行是标题行。
• 指定分隔符: 使用 delimiter 参数可以指定 CSV 文件中的分隔符。例如，csv.reader(csvfile, delimiter=';') 会将分号作为分隔符。
• 指定引号字符: 使用 quotechar 参数可以指定 CSV 文件中的引号字符。例如，csv.writer(csvfile, quotechar='|') 会使用竖线作为引号字符。

总结

Python 的 csv 库提供了丰富的功能，可以帮助你轻松地处理 CSV 文件。通过使用 reader 和 writer 对象，你可以读取、写入和操作 CSV 文件中的数据。希望这份文档能帮助你更好地理解和使用 Python 处理 CSV 文件。