如何使Pandas Styler的样式应用包含列表头的整列？

你遇到的问题很常见：style.apply()默认只作用于表格的数据区域（也就是表头下方的单元格），表头属于单独的<thead>区域，所以不会被apply()的样式影响。要实现整列（包括表头）都应用背景色，我们需要分别处理数据单元格和表头单元格，结合apply()和set_table_styles()来完成。

完整解决方案代码 #

假设你已经定义了added_columns和dropped_columns两个列名列表，我们可以这样修改代码：

import pandas as pd

# 假设你的原始DataFrame是old，added_columns/dropped_columns已定义
# added_columns = ['Header 2']
# dropped_columns = ['需要标红的列名']

def highlight_col(x):
    # 处理数据区域的单元格
    if x.name in added_columns:
        return ['background-color: #67c5a4'] * x.shape[0]
    elif x.name in dropped_columns:
        return ['background-color: #ff9090'] * x.shape[0]
    else:
        # 用transparent代替None，避免样式冲突
        return ['background-color: transparent'] * x.shape[0]

# 构建表头的样式规则
table_styles = []
# 获取每列的位置索引（CSS选择器中nth-child从1开始计数）
col_positions = {col: idx + 1 for idx, col in enumerate(old.columns)}

# 给新增列的表头设置绿色背景
for col in added_columns:
    if col in col_positions:
        table_styles.append({
            'selector': f'th.col_heading:nth-child({col_positions[col]})',
            'props': [('background-color', '#67c5a4')]
        })

# 给删除列的表头设置红色背景
for col in dropped_columns:
    if col in col_positions:
        table_styles.append({
            'selector': f'th.col_heading:nth-child({col_positions[col]})',
            'props': [('background-color', '#ff9090')]
        })

# 组合应用样式：先处理数据单元格，再设置表头样式
styled_df = old.style.apply(highlight_col, axis=0).set_table_styles(table_styles)

# 导出到Excel（需要openpyxl引擎）
styled_df.to_excel('styled_output.xlsx', engine='openpyxl', index=False)

关键原理说明 #

1. 数据区域处理：style.apply(highlight_col, axis=0)仍然负责给表头下方的单元格设置背景色，axis=0表示按列应用样式。
2. 表头区域处理：set_table_styles()通过CSS选择器定位表头的<th>元素：
   • th.col_heading匹配所有列表头单元格
   • nth-child({col_positions[col]})精准定位到指定列的表头（避免列名重复或含特殊字符时的匹配问题）
3. 样式一致性：确保表头和数据单元格使用完全相同的颜色值，这样整列的视觉效果就统一了。

注意事项 #

• 避免使用background-color: None，改用transparent更符合CSS规范，防止出现样式异常。
• 如果你的列名包含特殊字符（比如空格、括号），使用nth-child定位比:contains()更可靠，后者可能因字符转义问题失效。

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