Python自动化拆分Excel表格

import os
from datetime import datetime
import openpyxl

def split_excel_file(file_path):
    '''
    根据第一列的内容拆分Excel表格，并将每个工作表保存为独立的文件。

    Args:
        file_path (str): Excel文件的路径。
    '''
    try:
        # 打开Excel文件
        wb = openpyxl.load_workbook(file_path)
        sheet = wb.worksheets[0]  # 获取第一个工作表

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

        # 获取第一列所有单元格的内容
        col1_values = [cell.value for cell in sheet['A']]
        # 排除A1单元格
        col1_values = col1_values[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 人)'])

        # 保存为新的excel文件
        new_file_path = os.path.splitext(file_path)[0] + '_拆分结果.xlsx'
        wb.save(new_file_path)

        # 获取所有工作表
        sheets = wb.sheetnames
        # 排除第一个工作表
        sheets = sheets[1:]

        # 创建拆分结果文件夹
        folder_name = '拆分结果'
        desktop_path = os.path.join(os.path.expanduser('~'), 'Desktop')
        folder_path = os.path.join(desktop_path, folder_name)
        if not os.path.exists(folder_path):
            os.mkdir(folder_path)

        # 遍历每个工作表
        for sheet_name in sheets:
            # 创建新的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(folder_path, file_name)
            new_wb.save(file_path)

            # 关闭新的Excel文件
            new_wb.close()

        # 关闭原Excel文件
        wb.close()

        print('拆分完成！！')

    except Exception as e:
        print(f'发生错误：{e}')

if __name__ == '__main__':
    file_path = r'C:\Users\Administrator\Desktop\工作簿1.xlsx'  # 请修改为您的文件路径
    split_excel_file(file_path)