python - CSV插入列追加列的方法

网上搜索了一个下午，终于做出来了比较适合的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')