【python】使用csv庫以字典格式讀寫csv文件

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'])

運行之后結果：