盘点一个Python自动化办公需求——多个压缩包中Excel的合并

Python进阶者

发布于 2023-08-31 08:46:48

1680

发布于 2023-08-31 08:46:48

文章被收录于专栏：Python爬虫与数据挖掘

一、前言

这个事情还得从前几天在Python星耀群【Wing】问了一个Python自动化办公处理的问题，需求倒是不难，一起来看看吧。

她的需求如下：我有多个压缩文件（zip格式），每个文件里面有3个excel表、表名是一样的。请教：如何解压文件，并将文件中3个excel合并成一个表呢？

二、实现过程

这里我找她要了3个压缩包文件，后来【狂吃山楂片】给了一个答案，代码如下：

from pathlib import Path
from zipfile import ZipFile
import pandas as pd


zip_path = Path('你的压缩文件目录').glob('*.zip')  # 只获取zip后缀的压缩文件
to_path = Path('你的目标目录')

# 逐个读取目录中压缩文件
for file in zip_path:
    # 将一个压缩文件里面的excel文件合并成一个
    with ZipFile(file) as zipf:
        df = pd.concat(pd.read_excel(zipf.open(i)) for i in zipf.namelist())
        # 合并后的一个表保存到目标目录中
        df.to_excel(to_path.joinpath(f'{file.stem}.xlsx'), index=False)

运行之后，顺利地解决了粉丝的问题。