Pandas案例精进 | 续集：自动分割汇总表写入到子表

大家好！

欢迎来到「Pandas案例精进」专栏，点击橙字查看全部

前文回顾：Pandas案例精进 | 自动分割汇总表写入到子表中

本文是承接上一篇的实战案例，没看过的小伙伴建议先点击👆上方链接查看前文

Pandas案例需求

Pandas案例精进 | 续集：自动分割汇总表写入到子表

现在的新需求跟之前的区别有：

汇总表多了级别字段，需要根据不同的级别对应不同的文件夹：

Pandas案例精进 | 续集：自动分割汇总表写入到子表

所属区的xlsx文件有时可能是xls，并不一定是xlsx的。

各级别文件夹中存在一些不能匹配汇总表的垃圾文件需要删除。

汇总表中所有的对应项目并不是都在级别文件夹中存在，不存在的只提示哪些不存在，无需额外处理。

下面是我的实现过程：

数据加载

import os  
import pandas as pd  

excel_dir = os.getcwd()  

data = pd.read_excel(f"{excel_dir}/汇总.xlsx", sheet_name='明细')  
data

本文直接设置excel_dir为工作目录，大家如果代码和数据不在同一文件夹，可以根据自己的情况更改设置文件路径。

Pandas案例精进 | 续集：自动分割汇总表写入到子表

遍历计算出每级别所涉及的区

level_areas = {}  
for i, row in data.iterrows():  
    areas = level_areas.setdefault(row['级别'], set())  
    areas.add(row['所属区'])  
level_areas

执行结果

{'一级': {'B区'}, '二级': {'A区', 'C区'}, '三级': {'D区', 'E区'}}

删除多余文件+转换xls

这里会用到win32com.client库，需要额外安装。

pip install pywin32

也可使用国内清华源来加快Python库的安装速度。

pip install pywin32 -i https://pypi.tuna.tsinghua.edu.cn/simple

安装后使用

import win32com.client as win32  

for level in data['级别'].unique():  
    areas = level_areas[level]  
    files = os.listdir(f"{excel_dir}/{level}")  
    for file in files:  
        tag = file.replace(".xlsx", "").replace(".xls", "")  
        filename = f"{excel_dir}/{level}/{file}"  
        if not tag in areas:  
            print(f"删除文件：{filename}")  
            os.remove(filename)  
        elif file.endswith(".xls"):  
            print(f"将 {filename} 转换为 {filename}x")  
            excel = win32.gencache.EnsureDispatch('Excel.Application')  
            try:  
                wb = excel.Workbooks.Open(filename)  
                wb.SaveAs(f"{filename}x", FileFormat=51)  #FileFormat = 51 is for .xlsx extension  
                print("转换成功")  
            finally:  
                wb.Close()  #FileFormat = 56 is for .xls extension  
            excel.Application.Quit()  
            os.remove(filename)

上述代码可以实现删除级别文件夹多余的文件并将xls转换为xlsx。

比如将 一级/B区.xls 转换为 一级/B区.xlsx

注意：使用pywin32转换excel文件格式时，绝对路径的盘符后面的分隔符必须是反斜杠\，后面的路径分隔符用正斜杠或反斜杠都可以。

遍历出级别和区域

for (level, area), df in data.groupby(['级别', '所属区']):  
    print(level, area)  
    df = df.iloc[:, 2:]  
    display(df)

执行结果

Pandas案例精进 | 续集：自动分割汇总表写入到子表

写出结果

最后，利用openpyxl库将结果分别写入各级别、各Excel文件中。

from openpyxl import load_workbook  

for (level, area), df in data.groupby(['级别', '所属区']):  
    print(level, area)  
    df = df.iloc[:, 2:]  
    out_file_name = f"{excel_dir}/{level}/{area}.xlsx"  
    if not os.path.exists(out_file_name):  
        print(out_file_name, "文件不存在，跳过")  
        continue  
    print("准备写出到：", out_file_name)  
    workbook = load_workbook(filename=out_file_name)  
    sheet = workbook.active  
    # 先删除第4行之后的旧数据，预计1000行完全够用  
    sheet.delete_rows(idx=4, amount=1000)  
    # 然后再进行添加数据  
    for row in df.values.tolist():  
        sheet.append(row)  
        print(row)  
    print(f"保存到{out_file_name}文件中")  
    workbook.save(filename=out_file_name)  
    workbook.close()

Pandas案例精进 | 续集：自动分割汇总表写入到子表