最近遇到给数据集打标签的问题,查了网上相关CSV追加写入列的方法,很多都是COPY不起作用。简单问题复杂化了。
下面几句代码搞定:
label = [] #这是需要追加写入列的数据列表
csv_data = pd.read_csv('example.csv', low_memory = False)#example.csv是需要被追加的CSV文件,low_memory防止弹出警告
csv_df = pd.DataFrame(csv_data)
csv_df['label'] = label
csv_df.to_csv('out_csv.csv',index = None) #追加后生成的文件为out_csv