如何将含指定字段的DataFrame导出为可按group_name折叠的Excel表格

要实现你需要的按group_name列展开/折叠的Excel表格效果，可以借助pandas结合xlsxwriter引擎来完成，具体步骤如下：

核心思路 #

• 先对DataFrame按group_name字段排序，确保同一分组的记录连续排列
• 使用xlsxwriter作为导出引擎，利用它支持Excel分组（outline）的特性，将每个分组的明细行设置为可折叠状态
• 对group_name列的重复值进行处理，只保留每组第一行的分组名称，其余行该列留空，匹配你预期的样式

代码实现 #

import pandas as pd

# 假设你的DataFrame名为df，先按group_name排序
df_sorted = df.sort_values(by='group_name').reset_index(drop=True)

# 处理group_name列：每组仅第一行保留名称，其余为空
df_sorted['group_name'] = df_sorted['group_name'].where(df_sorted['group_name'] != df_sorted['group_name'].shift(), '')

# 创建Excel写入对象，指定xlsxwriter引擎
writer = pd.ExcelWriter('grouped_customers.xlsx', engine='xlsxwriter')
df_sorted.to_excel(writer, sheet_name='Sheet1', index=False)

# 获取工作表对象和工作簿对象
workbook = writer.book
worksheet = writer.sheets['Sheet1']

# 定义分组：遍历每个group_name的行范围，设置明细行的outline级别
current_group = None
start_row = 1  # 表头行是第0行，数据从第1行开始
for row_num, group in enumerate(df_sorted['group_name'], start=1):
    if group != '':
        if current_group is not None:
            # 对之前的分组设置明细行的outline级别（级别1，可折叠）
            for r in range(start_row, row_num):
                worksheet.set_row(r, None, None, {'level': 1, 'hidden': True})
        current_group = group
        start_row = row_num + 1
# 处理最后一个分组
if current_group is not None:
    for r in range(start_row, df_sorted.shape[0] + 1):
        worksheet.set_row(r, None, None, {'level': 1, 'hidden': True})

# 设置第一列（group_name列）的outline级别，让折叠按钮显示在列上方
worksheet.set_column(0, 0, None, None, {'level': 1})

# 启用Excel的自动折叠功能，设置分组样式
worksheet.outline_settings(True, True, True, True)

# 保存文件
writer.close()

关键说明 #

• df_sorted['group_name'].where(...)：通过对比当前行与上一行的group_name值，实现每组仅首行显示分组名称，其余行留空
• worksheet.set_row(..., {'level':1, 'hidden':True})：将明细行设置为级别1的分组，并默认隐藏，点击分组名称旁的折叠按钮即可展开/收起
• worksheet.outline_settings()：配置分组的显示样式，确保折叠按钮正常显示且交互符合预期

内容的提问来源于stack exchange，提问作者hjun