Python程序中用csv模块来操作csv文件的基本使用教程

当我们需要处理一些表格数据时，CSV文件类型是应用最广泛的一种格式之一。Python中提供了CSV模块，可以方便地读写CSV文件。

1. CSV模块的介绍

CSV模块提供的函数可以帮助我们方便地处理CSV文件，将表格数据读取到Python中进行操作，也可以将外部数据保存为CSV文件。

CSV模块中常用的函数有：

• csv.reader(csvfile, dialect='excel', **fmtparams)：读取CSV文件的函数，返回的是一个迭代器（iterator），可以用for循环逐行遍历文件内容。
• csv.writer(csvfile, dialect='excel', **fmtparams)：写入CSV文件的函数，将一串数据按照要求写入CSV文件中。
• csv.DictReader(csvfile, fieldnames=None, restkey=None, restval=None, dialect='excel', **fmtparams)：使用字典形式处理CSV文件，可以根据表头将表格的一行转换为字典。

2. CSV模块的读取

我们首先需要使用open()函数打开CSV文件，接着使用csv模块的reader()函数读取，这个函数返回的是一个迭代器，可以用for循环遍历文件内容。示例如下:

import csv

with open('example.csv') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row)

这个示例会输出CSV文件中每行的内容。

3. CSV模块的写入

写入CSV文件需要用到csv模块的writer()函数。我们先使用open()函数打开一个文件，接着使用writerow()函数将数据按要求写入文件。

import csv

with open('example.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['Name', 'Age'])
    writer.writerow(['Tom', '18'])
    writer.writerow(['John', '22'])

这个示例会创建一个example.csv文件，并把表格数据写入文件中。

4. CSV模块的操作字典

我们也可以使用字典的方式来处理CSV文件，这种方法可以根据表头将表格的一行转换为字典。使用DictReader()函数读取CSV文件，writer()函数写入CSV文件同理。

import csv

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

这个示例会输出CSV文件中每行的Name和Age两列内容，以字典形式呈现。

除了以上介绍的基本用法以外，CSV模块还提供了更多的函数可以进行高级操作。需要使用时可以查看Python官方文档或使用help(csv)命令获取使用文档。