按班级提取并生成到指定文件夹

import os
import pandas as pd

pd.set_option('display.max_columns', None)  # 显示所有列
pd.set_option('display.max_rows', None)  # 显示所有行


file_new = r'C:\zm\大三下\班级事务\所有班级'  # 准备生成文件的父目录
file_path = r'C:\zm\大三下\班级事务\2020-2021-2选修课教材领用'  # 被读取的xlsx文件父目录
xlsx_names = os.listdir(file_path)  # 读取   被读取的xlsx文件父目录 的文件名

all_class = set()      # 所有班级

# for k, xlsx_name in enumerate(xlsx_names):
#     data = pd.read_excel(file_path + '\\' + xlsx_name)
#     for i in data['行政班']:
#         all_class.add(i)
# print(all_class)


for k, xlsx_name in enumerate(xlsx_names):
    data = pd.read_excel(file_path + '\\' + xlsx_name)
    for i in all_class:
        data_new = data[data['行政班'] == f'{i}']  # 取出xx班级
        data_new.reset_index()
        if data_new.empty:	# 如果没有改班级进行下一判断
            continue
        data_new.to_excel(file_new + '\\' + f'{i}' + '\\' + xlsx_name)  #r 如果有则生成文件