Python系列（26）—— 导入CSV

一、csv模块的基本用法

CSV（Comma-Separated Values）文件是一种常见的数据交换格式，它使用逗号作为字段之间的分隔符，并以纯文本形式存储表格数据。

Python的csv模块提供了一个简单易用的接口来处理CSV文件。要使用csv模块，首先需要导入它，然后可以使用csv.reader()或csv.DictReader()函数来读取CSV文件。

1. 使用csv.reader()

csv.reader()函数返回一个读取器对象，该对象可以迭代CSV文件的每一行。每一行都被解析为一个列表，其中每个元素对应CSV文件中的一个字段。

示例：

import csv # 打开CSV文件 with open('path/to/your/file.csv', 'r') as file: # 创建CSV读取器 reader = csv.reader(file) # 迭代读取每一行 for row in reader: print(row)

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10

在上面的示例中，open()函数用于打开CSV文件，并返回一个文件对象。csv.reader()函数接受这个文件对象作为参数，并返回一个读取器对象。然后，我们可以使用for循环迭代读取器对象，每次迭代都会得到一个包含当前行字段的列表。

2. 使用csv.DictReader()

csv.DictReader()函数类似于csv.reader()，但它将CSV文件的每一行解析为一个字典，其中字典的键是CSV文件的列标题，值是对应字段的值。

示例：

import csv # 打开CSV文件 with open('path/to/your/file.csv', 'r') as file: # 创建CSV字典读取器 reader = csv.DictReader(file) # 迭代读取每一行 for row in reader: print(row)

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10

在这个示例中，csv.DictReader()函数将CSV文件的每一行解析为一个字典，使得我们可以使用列标题作为键来访问字段的值。这通常比使用csv.reader()更方便，特别是当CSV文件包含多列且列标题具有描述性时。

二、处理CSV文件中的特殊字符和格式

有时，CSV文件可能包含特殊字符（如引号、换行符等）或特殊的格式（如日期格式）。在这些情况下，我们可能需要更复杂的处理。

1. 处理特殊字符

默认情况下，csv模块会自动处理CSV文件中的特殊字符，如将包含特殊字符的字段用引号括起来。但是，如果你需要更精细的控制，可以使用csv.reader()或csv.DictReader()的quoting、escapechar等参数来指定如何处理特殊字符。

2. 处理日期格式

如果CSV文件中的日期字段是字符串格式，你可能需要将其转换为Python的日期对象。这可以通过使用datetime模块中的函数来实现。

示例：

import csv from datetime import datetime # 假设CSV文件的日期格式是 '%Y-%m-%d' date_format = '%Y-%m-%d' # 打开CSV文件 with open('path/to/your/file.csv', 'r') as file: reader = csv.DictReader(file) # 迭代读取每一行，并将日期字符串转换为日期对象 for row in reader: date_str = row['date_column'] # 假设日期字段的列标题是 'date_column' date_obj = datetime.strptime(date_str, date_format) row['date_column'] = date_obj print(row)

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12
• 13
• 14
• 15
• 16

在上面的示例中，我们使用datetime.strptime()函数将日期字符串转换为日期对象。你需要根据CSV文件中实际的日期格式来指定date_format变量的值。

三、注意事项

• 确保CSV文件的路径是正确的，并且文件可以被Python程序访问。
• 如果CSV文件很大，使用with语句来打开文件是一个好习惯，它可以确保文件在操作完成后被正确关闭。
• 根据CSV文件的具体内容和需求选择合适的读取方式（csv.reader()或csv.DictReader()）。
• 如果CSV文件中包含特殊字符或特殊的格式，你可能需要编写额外的代码来处理这些情况。

总结：Python的csv模块提供了强大的功能来导入和处理CSV文件。通过掌握csv.reader()和csv.DictReader()的基本用法，以及如何处理