【附代码】Python Excel合并单元格(OpenPyXL) Pandas.DataFrame groupby样式保存xlsx
import openpyxl
from openpyxl.styles import PatternFill
# 假设df是一个Pandas DataFrame,其中包含需要合并的行
# 假设df有'A'列,我们将根据'A'列的值进行分组,并合并相应的单元格
def merge_cells_in_column(sheet, column, start_row, end_row, fill=None):
# 在指定列的指定行范围内合并单元格
for row in sheet[column][start_row:end_row]:
for cell in row:
if fill:
cell.fill = fill
cell.merge = True
def groupby_and_merge(df, column_to_groupby, sheet_name='Sheet1', fill=None):
# 创建一个新的Excel工作簿
wb = openpyxl.Workbook()
ws = wb.active
ws.title = sheet_name
# 根据groupby列的值进行分组并保存
for group, data in df.groupby(column_to_groupby):
# 将数据写入工作表
for row_index, row in data.iterrows():
for col_index, value in row.items():
ws.cell(row=row_index+1, column=col_index+1).value = value
# 合并相同的组名所在的单元格
start_row = 1 # 从第一行开始
end_row = len(data) + 1 # 到数据结束的下一行
merge_cells_in_column(ws, column_to_groupby, start_row, end_row, fill)
# 保存工作簿
wb.save('merged_cells.xlsx')
# 示例DataFrame和groupby调用
# df = pandas.DataFrame({'A': ['foo', 'bar', 'foo', 'bar'], 'B': [1, 2, 3, 4]})
# groupby_and_merge(df, 'A')
这段代码定义了两个函数:merge_cells_in_column
和 groupby_and_merge
。merge_cells_in_column
函数负责合并指定列的单元格,而 groupby_and_merge
函数则根据Pandas DataFrame中的某一列进行分组,并对每个组名所在的单元格进行合并。在示例DataFrame和groupby调用部分,可以通过更改DataFrame和列名来实现不同的合并需求。
评论已关闭