网上搜索了一个下午,终于做出来了比较适合的CSV文件插入列和追加列比较好用的方法
该方法首次需要在文件中有对应的数据,然后按照对应行,否则会报错
# 按列写入csv
# 参数一 准备写入的数据 ['title','data1','data2',...]
# 参数二 首次写入的数据 ['title','data1','data2',...]
def write_excel(datas, first=False):
if first:
rows = zip(datas)
with open('test1.csv', "w", newline='') as f:
writer = csv.writer(f)
for row in rows:
writer.writerow(row)
else:
with open('test1.csv', 'r') as csvFile:
rows = csv.reader(csvFile)
with open('test2.csv','w', newline='') as f:
writer = csv.writer(f)
i = 0
for row in rows:
row.append(datas[i])
i = i + 1
writer.writerow(row)
# 删除原表和重命名
os.remove('test1.csv')
os.rename('test2.csv','test1.csv')