Python3 CSV文件读写

写入

1. 列表写入

import csv
headers = ['id','name','age','level','score']
rows = [
        [1,'张三丰','48','宗师',23],
        [2,'李白','64','秀才',22],
        [3,'秦始皇','42','皇帝',21],
        [4,'霍元甲','32','武状元',21]
    ]
# newline参数，避免csv文件出现空行
with open('test.csv','w',newline='') as f:
    f_csv = csv.writer(f)
    f_csv.writerow(headers)
    f_csv.writerows(rows)

2. 字典写入

import csv
headers = ['id','name','age','level','score']
rows = [
        {'id':1,'name':'张三丰','age':48,'level':'宗师','score':'23'},
        {'id':2,'name':'李白','age':64,'level':'秀才','score':'22'},
        {'id':3,'name':'秦始皇','age':42,'level':'皇帝','score':'21'},
        {'id':4,'name':'霍元甲','age':32,'level':'武状元','score':'21'}
    ]
# 如果存入乱码，添加 encoding='utf-8-sig'
with open('test2.csv','w',newline='')as f:
    f_csv = csv.DictWriter(f,headers)
    f_csv.writeheader()
    f_csv.writerows(rows)

3. 写入结果

读取

import csv
with open('test.csv')as f:
    f_csv = csv.reader(f)
    for row in f_csv:
        print(row)