python学习：csv文件与字典操作

 

从csv文件读取字典数据。

使用DictReader可以像操作字典那样获取数据，把表的第一行（一般是标头）作为key。

csv文件内容：

name,area
俄罗斯,1707
加拿大,997
中国,960
美国,936

python代码：

import csv

filename = 'd:/0tixiao/country.csv'
with open(filename) as f:
    reader = csv.DictReader(f)
    for row in reader:
        # Max TemperatureF是表第一行的某个数据，作为key
        country_name = row['name']
        print(country_name)     

显示结果：

俄罗斯
加拿大
中国
美国

向csv文件写入字典数据：

import csv

headers = ['country', 'area']

datas = [
        {'country':'俄罗斯', 'area':1707},
        {'country':'加拿大', 'area':997},
        {'country':'中国', 'area':960},
        {'country':'美国', 'area':936}
        ]

filename = 'd:/0tixiao/country_w.csv'
with open(filename, 'w', newline='') as f:
    
    writer = csv.DictWriter(f, headers)
    writer.writeheader()
    
    for row in datas:
        writer.writerow(row)

打开 d:/0tixiao/country_w.csv 文件：

country,area
俄罗斯,1707
加拿大,997
中国,960
美国,936