合并文件夹下文件
import pandas as pd
import os
# 读取路径
path = r"G:\test"
# 写入文件名称
final_name = "final_name"
# 写入文件路径
path_ = r"G:\test"
# 声明一个空的DataFrame，用来做最终的数据合并
final_data = pd.DataFrame()
def get_all_files(path):
global final_data
files = os.listdir(path)
for file in files:
if os.path.isfile(path + "/" +file):
filename,extension=os.path.splitext(file)
# 判断文件类别
if extension == ".txt" :
print("读取"+filename+"文件中的内容...........")
data = pd.read_table(path+'/' +file)
final_data = final_data.append(data,ignore_index=True)
print(data)
elif extension=='.xlsx':
print("读取"+filename+"文件中的内容...........")
data = pd.read_excel(path+'/' +file)
final_data = final_data.append(data,ignore_index=True)
print(data)
elif extension=='.csv':
data = pd.read_csv(path +'/'+file)
final_data = final_data.append(data,ignore_index=True)
print(data)
# 判断是不是文件夹
elif os.path.isdir(path+'/'+file):
get_all_files(path + '/' + file)
get_all_files(path)
print("数据合并完成")
# 保存文件
final_data.to_csv(path_ + "/" + final_name + ".csv")
print("写入完成")
