CSV(Comma Separated Values)是一种常见的电子表格文件格式,它使用逗号分隔不同字段,每行表示一条记录。Python的CSV模块提供了一种方便的方式来处理CSV文件。本文将从多个角度介绍Python CSV模块的使用实例,包括读取CSV文件、写入CSV文件、处理CSV数据等。

1. 读取CSV文件

Python CSV模块使用实例

使用Python CSV模块读取CSV文件非常简单,只需要使用csv.reader()函数即可。下面是一个读取CSV文件的示例代码:

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

在上面的代码中,我们打开一个名为data.csv的文件,并使用csv.reader()函数来读取它。delimiter参数指定了字段之间的分隔符,quotechar参数指定了字段的引号字符。在使用reader对象迭代时,每次迭代返回一个包含该行数据的列表。

2. 写入CSV文件

使用Python CSV模块写入CSV文件也非常简单,只需要使用csv.writer()函数即可。下面是一个写入CSV文件的示例代码:

import csv
with open('data.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL)
writer.writerow(['Name', 'Age', 'Gender'])
writer.writerow(['John Smith', 25, 'Male'])
writer.writerow(['Jane Doe', 30, 'Female'])

在上面的代码中,我们打开一个名为data.csv的文件,并使用csv.writer()函数来写入它。在使用writer对象写入数据时,我们可以使用writerow()函数来写入一行数据。在写入数据之前,我们需要使用writerow()函数写入表头。

3. 处理CSV数据

使用Python CSV模块处理CSV数据也非常方便。下面是一个处理CSV数据的示例代码:

import csv
with open('data.csv', newline='') as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
print(row['Name'], row['Age'], row['Gender'])

在上面的代码中,我们使用csv.DictReader()函数来读取CSV文件,并将每一行数据转换为一个字典。在使用reader对象迭代时,每次迭代返回一个包含该行数据的字典。我们可以使用字典的键来访问每个字段的值。

除了上述示例代码外,Python CSV模块还提供了许多其他的函数和选项,可以更加灵活地处理CSV文件。例如,我们可以使用csv.field_size_limit()函数来设置CSV字段的最大长度,使用csv.register_dialect()函数来注册一个自定义的CSV方言等。