如何识别DataFrame列中的分组并为其添加外边框？

原始数据与需求 #

首先定义目标DataFrame：

import pandas as pd

df = pd.DataFrame({'class': ['class_a', 'class_a', 'class_a', 'class_b', 'class_c', 'class_c'],
 'name': ['name_1', 'name_2', 'name_3', 'name_1', 'name_1', 'name_2'],
 'id': [5, 7, 1, 2, 3, 8]})

需求：为class列的每个分组绘制完整的蓝色实线矩形边框。

原方案的问题 #

原方案仅给每个分组首行添加了顶部蓝色边框，导致：

• 分组缺失底部边框，无法形成完整矩形
• 整个表格没有外边框，视觉上不完整

原方案代码：

s = df.style
for idx, group_df in df.groupby('class'):
    s = s.set_table_styles({group_df.index[0]: [{'selector': '', 'props': 'border-top: 3px solid blue;'}]}, 
                       overwrite=False, axis=1)

修复缺失外边框的解决方案 #

修改代码，为每个分组的首行添加上边框、末行添加下边框，同时给整个表格添加外边框，形成完整的分组矩形效果：

import pandas as pd

df = pd.DataFrame({'class': ['class_a', 'class_a', 'class_a', 'class_b', 'class_c', 'class_c'],
 'name': ['name_1', 'name_2', 'name_3', 'name_1', 'name_1', 'name_2'],
 'id': [5, 7, 1, 2, 3, 8]})

# 初始化样式对象
s = df.style

# 1. 给整个表格添加外边框，确保最外层的矩形
s = s.set_table_styles([
    {'selector': 'table', 'props': 'border: 3px solid blue;'}
], overwrite=False)

# 2. 为每个分组添加上下边框，区分内部分组
for _, group_df in df.groupby('class'):
    # 分组首行添加上边框
    s = s.set_table_styles({
        group_df.index[0]: [{'selector': '', 'props': 'border-top: 3px solid blue;'}]
    }, overwrite=False, axis=1)
    # 分组末行添加下边框
    s = s.set_table_styles({
        group_df.index[-1]: [{'selector': '', 'props': 'border-bottom: 3px solid blue;'}]
    }, overwrite=False, axis=1)

# 查看效果
display(s)

效果说明 #

• 整个表格拥有完整的蓝色外边框
• 每个class分组被独立的蓝色矩形框包裹，上下边框与外边框衔接，形成清晰的分组视觉区分

注：关于保存到Excel样式丢失的问题，Pandas的style样式仅支持HTML渲染，无法直接导出到Excel。若需要导出带样式的Excel，需使用openpyxl或xlsxwriter库手动设置单元格边框样式，这部分属于另一个实现场景。

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