使用 Python Openpyxl 拆分 Excel 表格

本文将介绍如何利用 Python 3.8 版本的 openpyxl 库对 Excel 表格进行拆分,具体操作是根据原工作表第一列单元格的内容创建新的工作表,并复制相关内容。

代码示例

import openpyxl

# 打开 Excel 文件
wb = openpyxl.load_workbook(r'C:\Users\Administrator\Desktop\工作簿1.xlsx')
sheet = wb.worksheets[0]  # 获取第一个工作表
sheet.freeze_panes = 'A2'  # 冻结 A1 窗格

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

# 获取第一列所有单元格的内容
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 文件
wb.save(r'C:\Users\Administrator\Desktop\拆分表格结果.xlsx')

代码解析

  1. 导入 openpyxl 库: import openpyxl
  2. 打开 Excel 文件: wb = openpyxl.load_workbook(r'C:\Users\Administrator\Desktop\工作簿1.xlsx')
  3. 获取第一个工作表: sheet = wb.worksheets[0]
  4. 冻结 A1 窗格: sheet.freeze_panes = 'A2'
  5. 获取第二行所有单元格的内容:
    row2_values = []
    for cell in sheet[2]:
        row2_values.append(cell.value)
    print(row2_values)
    
  6. 获取第一列所有单元格的内容:
    col1_values = [cell.value for cell in sheet['A']]
    # 排除 A1 单元格
    col1_values = col1_values[1:]
    
  7. 拆分第一列单元格内容到新的工作表:
    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')
    
  8. 删除工作表名字为表头(1 人)的工作表:
    if '表头 (1 人)' in wb.sheetnames:
        wb.remove(wb['表头 (1 人)'])
    
  9. 保存为新的 excel 文件: wb.save(r'C:\Users\Administrator\Desktop\拆分表格结果.xlsx')

总结

本代码通过 openpyxl 库实现了对 Excel 表格的拆分功能,根据第一列单元格的内容创建新的工作表,并复制相关内容。用户可以根据自己的需求修改代码,实现更多功能。

注意: 由于 openpyxl 库不支持对 Excel 文件的直接写入操作,因此需要先将数据读入内存,再进行修改,最后再保存到新的 Excel 文件。

Python Openpyxl 拆分表格:根据第一列内容创建新工作表

原文地址: https://www.cveoy.top/t/topic/fXcx 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录