【python】使用csv库以字典格式读写csv文件

西西嘛呦

发布于 2020-08-26 10:34:50

1.7K00

代码可运行

文章被收录于专栏：数据分析与挖掘数据分析与挖掘

运行总次数：0

代码可运行

1、使用csv.DictWriter()写入字典格式的数据

import csv
    

with open('test.csv', 'w', newline='') as csvfile:
    fieldnames = ['first_name', 'last_name']
    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)

    writer.writeheader()
    writer.writerow({'first_name': 'Baked', 'last_name': 'Beans'})
    writer.writerow({'first_name': 'Lovely', 'last_name': 'Spam'})
    writer.writerow({'first_name': 'Wonderful', 'last_name': 'Spam'})

运行之后结果：

2、使用csv.DictReader()读取字典格式的数据

with open('test.csv', 'r') as csvfile:
    #fieldnames = ['first_name', 'last_name']
    reader = csv.DictReader(csvfile)
    for row in reader:
        print(row['first_name'],row['last_name'])

运行之后结果：