工作中经常遇到操作csv文件,读取csv文件相对比较简单,而写入csv文件时,经常会遇到数据没有表头(col_name)的情况,这时就需要自己定义并写入表头。
由于pyhton程序是按顺序从对到尾执行的,所以对csv文件表头按顺序先进行写入,而我们有时在生成csv文件时,也是一条一条解析出来的,所以在做csv文件表头时,也需要在解析保存数据前,做为一行数据进行写入,然后再对文件进行保存的方法进行。
主要代码如下:
import csv
yuwen_list = [['wanger','3-2','68'],
['lisi','3-1','88'],
['wangwu','3-1','75'],
['maliu','3-3','90']]
# 写表头
with open('.\语文成绩.csv',mode='a',encoding='utf-8',newline='') as f:
csv_writer = csv.writer(f)
csv_writer.writerow(['序号','姓名','班级','语文'])
# 写数据
for i in range(len(yuwen_list)):
print(yuwen_list[i])
# 逐行提取数据
name = yuwen_list[i][0]
gread = yuwen_list[i][1]
score = yuwen_list[i][2]
# 将数据追加入文件
with open('.\语文成绩.csv',mode='a',encoding='utf-8',newline='') as f:
csv_writer = csv.writer(f)
csv_writer.writerow([i+1,name,gread,score])
f.close()
生成的csv文件是下面这种样式: