Python CSV读写字典数据

本文创建时间: 2022.07.20 22:22:30，更新时间: 2022.07.20 22:26:43

生成数据

import csv
from random import randint

raw_data = [{'a': randint(1, 10), 'b': randint(10, 100),
             'c': randint(100, 1000)} for _ in range(6)]

print(f'Raw data:')
for data in raw_data:
    print(data)

写入字典数据到CSV

# write CSV
with open('./demo.csv', 'w') as f:
    fields = ['a', 'b', 'c']
    writer = csv.DictWriter(f, fields)
    writer.writeheader()
    for data in raw_data:
        writer.writerow(data)

从CSV文件读取字典数据

# read CSV
with open('./demo.csv', 'r') as f:
    print(f'Read from CSV:')
    reader = csv.DictReader(f)
    for row in reader:
        print(row)

参考

• csv --- CSV 文件读写 — Python 3.10.5 文档