csv模块是Python标准库中的一部分,它提供了一种从文件中读取和写入纯文本数据的方法。csv文件是一种简单的文本文件,它由一系列以逗号分隔的值组成,它们可以被用来存储表格数据,如电子表格或数据库中的数据。
csv模块的使用方法
使用csv模块可以方便地读取和写入csv文件,csv模块提供了两个类:csv.reader和csv.writer,分别用于读取和写入。
csv.reader类
csv.reader类可以用来读取csv文件。它接受一个参数,即要读取的文件对象,返回一个csv.reader对象,该对象的每一行都是一个列表,其中的每个元素都是该行中的一个字段。
import csv with open('data.csv', 'r') as f: reader = csv.reader(f) for row in reader: print(row)
csv.writer类
csv.writer类可以用来写入csv文件。它接受一个参数,即要写入的文件对象,返回一个csv.writer对象,该对象可以接受一个列表作为参数,并将其写入文件。
import csv data = [['name', 'age'], ['Tom', '20'], ['Bob', '25']] with open('data.csv', 'w') as f: writer = csv.writer(f) for row in data: writer.writerow(row)
csv模块的其他方法
除了csv.reader和csv.writer类之外,csv模块还提供了一些其他的方法,可以用来设置csv文件的格式,比如分隔符、引号和行结束符等。
- csv.Dialect:用于定义csv文件的格式,比如分隔符、引号和行结束符等。
- csv.excel:用于定义excel格式的csv文件。
- csv.excel_tab:用于定义tab分隔的csv文件。
- csv.register_dialect:用于注册一个新的csv格式。
- csv.unregister_dialect:用于注销一个已注册的csv格式。
- csv.get_dialect:用于获取已注册的csv格式。
csv模块的使用示例
下面是一个使用csv模块读取csv文件的示例:
import csv with open('data.csv', 'r') as f: reader = csv.reader(f, delimiter=';', quotechar='"') for row in reader: print(row)
上面的示例使用csv.reader类读取了一个csv文件,并使用delimiter和quotechar参数指定了文件的分隔符和引号。