Python pandas依列拆分为多个Excel文件

import pandas as pd
data = pd.read_excel("D:\yhd_python\yhd-python依列拆分Excel\汇总.xlsx")
rows = data.shape[0] #获取行数 shape[1]获取列数
print(rows)
data["身份证"]=[" %i"%i for i in data["身份证"]]
department_list=[]

for i in range(rows):
    temp=data["班别"][i]
    
    if temp not in department_list:
        department_list.append(temp)
for department in department_list:
    new_df=pd.DataFrame()
    
    for i in range(0,rows):
        if data["班别"][i]==department:
            new_df=pd.concat([new_df,data.iloc[[i],:]],axis=0,ignore_index=True)
    #print(new_df)        
    new_df.to_excel(str(department)+".xls", index = False)