Python CSV模块详解与实例

"这篇文档详细介绍了Python CSV模块的使用，通过多个实例展示了reader、writer、DictReader和DictWriter.register_dialect等方法的功能和用法。CSV模块是Python中用于处理逗号分隔值（CSV）文件的一个标准库，它提供了一种方便的方式来读写CSV文件。" Python CSV模块是Python标准库中的一个组件，专门用于处理CSV数据。CSV（Comma Separated Values）格式是一种常见的数据存储格式，常用于电子表格和数据库导入导出。CSV模块提供了多种方法来方便地读写CSV文件。 1. `reader`对象：`csv.reader`用于读取CSV文件。它接受一个可迭代对象，如打开的文件对象或包含CSV数据的列表。默认的分隔符是逗号，但可以通过`dialect`参数设置不同的分隔符，如制表符。`fmtparam`可以用来覆盖特定的格式参数。例如： ```python import csv with open('your.csv', 'r') as csvfile: reader = csv.reader(csvfile) for line in reader: print(line) ``` 2. `writer`对象：`csv.writer`用于写入CSV文件。同样需要一个可迭代对象，这里通常是打开的二进制文件对象。`writerow()`方法用于写一行数据，可以是列表或其他可迭代对象。例如： ```python import csv with open('your.csv', 'w', newline='') as csvfile: writer = csv.writer(csvfile) writer.writerow(['Column1', 'Column2', 'Column3']) lines = [[i, i+1, i+2] for i in range(3)] for line in lines: writer.writerow(line) ``` 3. `DictReader`：`csv.DictReader`与`reader`类似，但它将每一行读取为一个字典，键是CSV文件的第一行（假设是列名），值是后续行的对应列。这使得按列名访问数据变得容易。例如： ```python import csv with open('your.csv', 'r') as csvfile: dict_reader = csv.DictReader(csvfile) for row in dict_reader: print(row['Column1'], row['Column2'], row['Column3']) ``` 4. `DictWriter`：与`DictReader`相对应，`csv.DictWriter`允许你以字典形式写入CSV文件。你需要指定字段名，然后使用`writerows()`方法写入包含字典的列表。例如： ```python import csv fieldnames = ['Column1', 'Column2', 'Column3'] data = [{'Column1': i, 'Column2': i+1, 'Column3': i+2} for i in range(3)] with open('your.csv', 'w', newline='') as csvfile: dict_writer = csv.DictWriter(csvfile, fieldnames=fieldnames) dict_writer.writeheader() dict_writer.writerows(data) ``` 5. `register_dialect`和`list_dialects`：`csv.register_dialect`用于注册自定义的分隔符风格，以便在读写时使用。`list_dialects`则返回所有已注册的方言名称。 在处理CSV数据时，Python CSV模块提供了强大的功能，能够适应各种需求，无论是简单的数据交换还是复杂的转换操作。通过理解并熟练使用这些方法，开发者可以高效地处理CSV文件。