Python 可扩展文件数据合并

由于前一阵我需要合并excel文件数据我的excel2019加强版的查询功能莫名报错所以只能写个程序解决了
import pandas as pd
import os

def excel(file_dir , name):
    new_filename = file_dir + '\\' +name +'.xlsx'
    print("合并后的文件路径" + new_filename)
    file_list = os.listdir(file_dir)
    new_list = []
    print("正在合并excel文件数据!")
    for file in file_list:
        file_path = os.path.join(file_dir,file)
        dataframe = pd.read_excel(file_path)
        new_list.append(dataframe)
        print("正在获取 "+ file +" 的文件数据!")
    df = pd.concat(new_list)
    df.to_excel(new_filename,index=False)
    print("excel文件数据获取完成!!请注意查看!!")

def txt():
    pass

def choose():
    file_dir = input("请输入需要合并文件数据的目录绝对路径:")
    name = input("请输入合并后的文件名字(不用写后缀):")
    file_type = input("请输入要合并文件数据的文件类型后缀名:")
    if file_type == "txt" or file_type ==".txt":
        txt()
    elif file_type == "xls" or file_type == ".xls":
        excel(file_dir , name)
    else:
        print("该类型文件暂不支持合并数据!!")

if __name__ == "__main__":
    choose()
    input()

 

posted @ 2021-02-24 00:02  Ruler_Madman  阅读(53)  评论(0)    收藏  举报