主要实现功能:
在同一文件夹下的所有csv文件全部合并到同一个csv文件中,并将csv文件的表头保留
1 import os 2 import pandas as pd 3 path = os.getcwd() 4 files_list = os.listdir(path) 5 csv_list = [] 6 for f in files_list: 7 print(os.path.splitext(f)[1]) 8 if os.path.splitext(f)[1]=='.csv':#os.path.splitext()用法--分离文件名与扩展名 9 csv_list.append(path+'\\'+f) 10 else: 11 pass 12 save_name_data= open('全部数据.csv',mode='a+') 13 save_name = '全部数据.csv' 14 #保留表头 15 df = pd.read_csv(csv_list[0],encoding='utf-8') 16 df.to_csv(path+'\\'+save_name,encoding='utf-8',index=False,header=True,mode='a+') 17 #文件夹下所有csv文件第二行起,全部写入同一个csv文件中 18 for i in range(1,len(csv_list)): 19 df = pd.read_csv(csv_list[i],encoding='utf-8') 20 df.to_csv(path+'\\'+save_name,encoding='utf-8',index=False,header=False,mode='a+') 21 save_name_data.close()