Python Openpyxl 拆分表格:根据第一列内容创建新工作表
使用 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')
代码解析
- 导入 openpyxl 库:
import openpyxl - 打开 Excel 文件:
wb = openpyxl.load_workbook(r'C:\Users\Administrator\Desktop\工作簿1.xlsx') - 获取第一个工作表:
sheet = wb.worksheets[0] - 冻结 A1 窗格:
sheet.freeze_panes = 'A2' - 获取第二行所有单元格的内容:
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')
总结
本代码通过 openpyxl 库实现了对 Excel 表格的拆分功能,根据第一列单元格的内容创建新的工作表,并复制相关内容。用户可以根据自己的需求修改代码,实现更多功能。
注意: 由于 openpyxl 库不支持对 Excel 文件的直接写入操作,因此需要先将数据读入内存,再进行修改,最后再保存到新的 Excel 文件。
原文地址: https://www.cveoy.top/t/topic/fXcx 著作权归作者所有。请勿转载和采集!