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

大家好！

欢迎来到「Pandas案例精进」专栏

今天分享的是一个之前的案例，里面涉及的方法可能有些过时，但处理思想仍有较高的参考价值。

Pandas案例需求

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

整理了一下需求大概的意思，即根据汇总表的分区字段自动填入指定的分区文件中：

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

对于分区表的文件，例如A区.xlsx、B区.xlsx等，需要先将3行之后已经存在的数据删除后再进行写入。

B区.xlsx在自动填入后，结果如下：

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

其实初始需求非常简单，我们下面看看怎么做吧。

汇总表的数据情况

import pandas as pd  
data = pd.read_excel("汇总.xlsx", sheet_name='明细')  
data

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

对于B区的数据如何写入呢？

筛选出准备写入B区的数据

df = data[data["所属区"] == "B区"]  
df

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

df = df.iloc[:, 1:]  
df.values.tolist()

结果：

[[111111, nan, '张一', '招商银行', 576.0, nan, nan, nan, '河南省'],  
 [111112, nan, '张二', '招商银行', 576.0, nan, nan, nan, '湖南省'],  
 [111113, nan, '张三', '招商银行', 1392.0, nan, nan, nan, '河南省']]

覆盖写入到对应的分区文件

workbook = load_workbook(filename="B区.xlsx")  
sheet = workbook.active

先删除第4行之后的旧数据，预计1000行完全够用，具体数量根据自己的数据来大致估算。

sheet.delete_rows(idx=4, amount=1000)

然后再进行添加数据

for row in df.values.tolist():  
    sheet.append(row)  
workbook.save(filename="B区.xlsx")  
workbook.close()

查看名为B区的Excel：

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

遍历分区字段的简单办法

for area, df in data.groupby('所属区'):  
    print(area)  
    display(df)

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

完整代码

data = pd.read_excel("汇总.xlsx", sheet_name='明细')  

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

好了经过以上步骤，就成功完成任务了：

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