使用 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 文件。