pandas-多文件合并

最新推荐文章于 2024-08-21 14:05:40 发布

BRUIN.

最新推荐文章于 2024-08-21 14:05:40 发布

阅读量1k

点赞数

分类专栏： python数据处理和分析文章标签： excel python

本文链接：https://blog.csdn.net/I_I___LO_VE___YA/article/details/105330943

版权

python数据处理和分析专栏收录该内容

14 篇文章 0 订阅

订阅专栏

使用glob获取所有得excel文件，pandas读取文件，保存至同一个文件中

import glob
import os
import pandas as pd


class MergeExcel(object):
    def __init__(self, headers):
        # 需要提取的列名
        self.headers = headers
        # 创建一个数据空间用于存储数据
        self.datas = pd.DataFrame(columns=self.headers)

    def get_file_path(self, dir_path):
        # 递归获取所有excel文件
        file_paths = glob.glob(dir_path + '/' + '*.xls*', recursive=True)
        return file_paths

    def merge_excel(self, file_path):
        # 读取、合并文件
        data = pd.read_excel(file_path)
        self.datas = self.datas.append(data[self.headers])

    def run(self):
        # 获取当前文件夹名称
        dir_path = os.path.dirname(__file__)
        # dir_path = os.getcwd()

        # 获取当前文件夹下所有excel文件路径
        # dir_path = r'C:\Users\xiongrenyi\Desktop\zhiyuan'

        # 获取所有excel文件路径
        file_paths = self.get_file_path(dir_path)
        print(file_paths)

        # 遍历所有的excel文件路径
        for file_path in file_paths:
            self.merge_excel(file_path)
            print(file_path.split('\\')[-1], '已合并')

        # 保存
        self.datas.to_excel('new_file.xlsx', sheet_name='Sheet1', index=False, header=True)


if __name__ == '__main__':
    # 输入需要提取的列名
    heads = ['学号', '第一志愿']

    merge_excel = MergeExcel(heads)
    merge_excel.run()