python 对数据分类并写入不同工作表

需求：

 方法1、 循环插入表、写入值

import os
import pandas  as pd
import openpyxl

wb_in = 'E:\excel_vba\test.xlsx'
df = pd.read_excel(wb_in)

wb = openpyxl.load_workbook(wb_in)  # 加载工作表
new_sheet_name = df['班级'].drop_duplicates()
for k, name in enumerate(new_sheet_name):
    sheet = wb.create_sheet(title=name)  # 插入工作表，指定名称，index 参数默认为 -1
    
    # 写入表头
    for i, v in enumerate(df.columns):
        sheet.cell(row=1, column=i+1).value = v

    for i, row in enumerate(df[df['班级']==new_sheet_name[k]].values):
        for j, v in enumerate(row):
            sheet.cell(row=i+2, column=j+1).value = v # 第 2 行， 第 1 列开始写入数据
            
wb_out = os.path.join(os.path.dirname(wb_in), 'test_1.xlsx')
wb.save(wb_out)

方法2、直接保存 DataFrame

import os
import pandas  as pd
import openpyxl

wb_in = 'E:\excel_vba\test.xlsx'
wb_out = os.path.join(os.path.dirname(wb_in), 'test_1.xlsx')

df = pd.read_excel(wb_in)
new_sheet_names = df['班级'].drop_duplicates()
book = openpyxl.load_workbook(wb_in)

writer = pd.ExcelWriter(wb_out, engine='openpyxl')
writer.book = book

for name in new_sheet_names:
    df_a = df[df['班级']==name]
    df_a.to_excel(writer, sheet_name=name, index=False)
    
writer.save()