Python 使用 Openpyxl 拆分 Excel 文件

import openpyxl
import os
from datetime import datetime

# 定义全局变量
EXCEL_FILE_PATH = r'C:\Users\Administrator\Desktop\23春学生毕业情况表.xlsx'
OUTPUT_FOLDER_NAME = '拆分结果'

# 创建拆分结果文件夹
def create_output_folder(folder_name, desktop_path):
    folder_path = os.path.join(desktop_path, folder_name)
    if not os.path.exists(folder_path):
        os.makedirs(folder_path)
    return folder_path

# 拆分工作表
def split_worksheet(wb, sheet_name, output_folder_path):
    # 创建新的 Excel 文件
    new_wb = openpyxl.Workbook()
    new_sheet = new_wb.active

    # 复制工作表内容到新的 Excel 文件
    sheet = wb[sheet_name]
    for row in sheet.iter_rows():
        for cell in row:
            new_sheet[cell.coordinate].value = cell.value

    # 保存新的 Excel 文件
    date_str = datetime.now().strftime('%Y-%m-%d')
    file_name = f'{sheet_name}-{date_str}.xlsx'
    file_path = os.path.join(output_folder_path, file_name)
    new_wb.save(file_path)
    new_wb.close()

# 主函数
def main():
    # 打开 Excel 文件
    with openpyxl.load_workbook(EXCEL_FILE_PATH) as wb:
        sheet = wb.worksheets[0]
        sheet.freeze_panes = 'A2'

        # 获取第二行所有单元格的内容
        row2_values = [cell.value for cell in sheet[2]]

        # 获取第一列所有单元格的内容
        col1_values = [cell.value for cell in sheet['A']][1:]

        # 拆分第一列单元格内容到新的工作表
        unique_values = set(col1_values)
        for value in unique_values:
            count = col1_values.count(value)
            new_sheet_name = f'{value} ({count} 人)'
            new_sheet = wb.create_sheet(title=new_sheet_name)

            # 在新工作表的第一行插入原工作表A1单元格的内容
            new_sheet.insert_rows(1)
            for col_num, cell in enumerate(sheet[1], 1):
                new_sheet.cell(row=1, column=col_num, value=cell.value)

            # 添加第二行单元格内容到新工作表
            new_sheet.append(row2_values)

            # 添加第一列单元格内容到新工作表
            for row_num, cell_value in enumerate(col1_values):
                if cell_value == value:
                    new_sheet.append([cell.value for cell in sheet[row_num + 2]])

            # 删除新工作表中的空行
            for row in new_sheet.iter_rows():
                if all([cell.value is None for cell in row]):
                    new_sheet.delete_rows(row[0].row)

            # 合并第一行单元格并居中
            num_cols = new_sheet.max_column
            new_sheet.merge_cells(start_row=1, start_column=1, end_row=1, end_column=num_cols)
            new_sheet.cell(row=1, column=1).alignment = openpyxl.styles.Alignment(horizontal='center', vertical='center')

        # 删除工作表名字为表头（1）人的工作表
        if '表头 (1 人)' in wb.sheetnames:
            wb.remove(wb['表头 (1 人)'])

        # 获取所有工作表
        sheets = wb.sheetnames[1:]

        # 创建拆分结果文件夹
        desktop_path = os.path.join(os.path.expanduser('~'), 'Desktop')
        output_folder_path = create_output_folder(OUTPUT_FOLDER_NAME, desktop_path)

        # 遍历每个工作表并拆分
        for sheet_name in sheets:
            split_worksheet(wb, sheet_name, output_folder_path)

# 程序入口
if __name__ == '__main__':
    main()