Python DictReader读写csv文件

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/tomatomas/article/details/81005503

准备

此文档的示例代码是基于python3.5写的。
使用csv库前，先导入csv库：
import csv

读取内容

假设csv文件的内容如下图所示，DictReader会将第一行的内容（类标题）作为key值，第二行开始才是数据内容。即图中的csv文件有2列7行数据，第一列的key值为id，第二列的key值为class：

with open(‘name.csv’) as csvfile:
    reader = csv.DictReader(csvfile)
    for row in reader:
        #循环打印数据的id和class值，此循环执行7次
        print(row[‘id’],row[‘class’])

打印结果为：
0 5
1 4
2 13
…
6 15

写入内容

现假设我们要创建一个格式如上图所示的csv文件，则实现代码应该如下：

with open(‘name.csv’,’w’) as csvfile:
    writer = csv.DictWriter(csvfile,fieldnames=[‘id’,’class’])
    #写入列标题，即DictWriter构造方法的fieldnames参数
    writer.writeheader()
    for data in datas:
        writer.writerow({‘id’:data[0],’class’:data[1]})