Python Openpyxl 拆分Excel表格:根据特定列值创建新工作表
import openpyxl
import os
from datetime import datetime
from openpyxl import load_workbook
from openpyxl.styles import colors, Font, PatternFill, Alignment
from openpyxl.cell import Cell
# 打开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)
# 获取第一列所有单元格的内容
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:
row = [cell.value for cell in sheet[row_num + 2]]
new_row = []
for cell in row:
if isinstance(cell.value, str):
new_cell = Cell(new_sheet, column=cell.column, row=cell.row, value=None)
elif cell.value is None:
new_cell = Cell(new_sheet, column=cell.column, row=cell.row)
else:
new_cell = Cell(new_sheet, column=cell.column, row=cell.row, value=cell.value)
new_cell.font = cell.font
new_cell.fill = cell.fill
new_row.append(new_cell)
new_sheet.append(new_row)
# 删除新工作表中的空行
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 = Alignment(horizontal='center', vertical='center')
# 删除工作表名字为表头(1)人的工作表
if '表头 (1 人)' in wb.sheetnames:
wb.remove(wb['表头 (1 人)'])
# 保存为新的excel文件
wb.save(r'C:\Users\Administrator\Desktop\拆分表格结果.xlsx')
代码说明:
-
导入必要的库:
openpyxl用于操作 Excel 文件;os用于处理文件路径;datetime用于处理时间戳;colors、Font、PatternFill和Alignment用于设置单元格样式。 -
打开 Excel 文件: 使用
load_workbook()方法打开 Excel 文件,并获取第一个工作表。 -
获取数据: 获取第二行和第一列的所有单元格内容,并排除第一列的第一个单元格。
-
创建新工作表: 遍历第一列所有不同单元格内容,根据每个内容创建新的工作表,并将其命名为
单元格内容 (出现次数 人)。 -
复制数据到新工作表: 将原工作表的第一行和第二行内容复制到新工作表的第一行和第二行。并将第一列中与当前单元格内容相同的所有行的内容复制到新工作表。
-
删除空行: 删除新工作表中的空行。
-
合并单元格: 将新工作表的第一行单元格合并,并设置居中对齐。
-
删除工作表: 删除名为
表头 (1 人)的工作表。 -
保存新的 Excel 文件: 将修改后的 Excel 文件保存到指定路径。
修改说明:
在处理第一列单元格内容时,添加了对 cell.value 类型进行判断的逻辑,以避免出现 AttributeError: 'str' object has no attribute 'value' 错误。
if isinstance(cell.value, str):
new_cell = Cell(new_sheet, column=cell.column, row=cell.row, value=None)
elif cell.value is None:
new_cell = Cell(new_sheet, column=cell.column, row=cell.row)
else:
new_cell = Cell(new_sheet, column=cell.column, row=cell.row, value=cell.value)
new_cell.font = cell.font
new_cell.fill = cell.fill
这段代码会判断 cell.value 是否为字符串类型,如果是,则创建一个新的单元格对象,并将它的值设置为 None。如果不是字符串类型,则正常处理。
注意:
- 代码中的文件路径请根据实际情况进行修改。
- 代码中的
表头 (1 人)工作表名称也请根据实际情况进行修改。 - 确保已安装
openpyxl库。
使用说明:
- 将代码保存为
.py文件。 - 运行代码。
- 修改后的 Excel 文件将会保存在指定的路径。
原文地址: https://www.cveoy.top/t/topic/fXkf 著作权归作者所有。请勿转载和采集!