Python实现将Excel表格按某列拆分为多个sheet

实际数据分析中遇到需求，把某个Excel表格按照某一列分为多个sheet，并且要求如果某个key对应的行数较少应该合并到一个sheet中。

代码语言：Python复制

import pandas as pd
import bioquest as bq # https://jihulab.com/BioQuest/bioquest

从网上找随便了个数据做演示用

代码语言：Python复制

input_file=r"https://gitee.com/zhjx19/chaoyanghospital/raw/master/朝阳医院2018年销售数据.xlsx"
output_file=r"朝阳医院.xlsx"
key='商品名称'

读如数据，删除商品名称为na的行

代码语言：Python复制

data = pd.read_excel(input_file)
data.dropna(subset=key,inplace=True)

替换/为每，删除特殊字符（因为不能作为sheetname）

代码语言：Python复制

data.loc[:,key] = bq.st.replaces(string=data.loc[:,key],pattern=r"/",repl="每")
data.loc[:,key] = bq.st.replaces(string=data.loc[:,key],pattern=r"[\*?:/[]]",repl="")

如果某个key对应的行数少于50则合并在合并的药物这个sheet中，其他的key单独存在对应的sheet中

代码语言：Python复制

keys=data.loc[:,key].unique().tolist()
few_dict = {}
single_dict = {}
for i in keys:
    data_sub = data.groupby(key).get_group(i)
    if data_sub.shape[0]<50:
        few_dict[i] = data_sub
    else:
        single_dict[i] = data_sub

第一次写出合并的药物sheet

代码语言：Python复制

few = pd.concat(few_dict,ignore_index=True)
few.to_excel(output_file, sheet_name="合并的药物", index=False)

循环append sheet，最后close

代码语言：Python复制

writer = pd.ExcelWriter(output_file, engine='openpyxl',mode="a")
for k,v in single_dict.items():
    v.to_excel(writer, sheet_name=f"{k}", index=False)
writer.close()

python geopandas

0 人点赞