Pandas Styler applymap循环与分支代码结果不一致问题

问题场景 #

你原本用6个独立if分支调用applymap，能给对应列正确设置颜色：

df_style = df.style
if 'A' in df.columns:
    df_style = df_style.applymap(lambda x: 'color: red' if x > 0 else '', subset=['A'])
if 'B' in df.columns:
    df_style = df_style.applymap(lambda x: 'color: blue' if x > 0 else '', subset=['B'])
# ... 依次处理到Z列

但改成循环重构后，所有迭代列都被应用了最后一列（比如Z列）的颜色：

color_map = {'A': 'red', 'B': 'blue', ..., 'Z': 'green'}
df_style = df.style
for col, color in color_map.items():
    df_style = df_style.applymap(lambda x: f'color: {color}' if x > 0 else '', subset=[col])

核心原因 #

这是Python的闭包延迟绑定特性导致的。循环里定义的lambda表达式并没有在迭代时就捕获当前的color值，而是直到applymap实际执行（也就是Streamlit渲染DataFrame样式的时候）才会去查找color变量的当前值。循环结束后，color变量最终指向的是循环最后一次迭代的元素（比如Z列对应的green），所以所有lambda执行时都会使用这个最后值。

解决方法 #

方法1：用默认参数捕获当前值

默认参数会在lambda定义时就完成求值，能固定住当前循环的color：

color_map = {'A': 'red', 'B': 'blue', ..., 'Z': 'green'}
df_style = df.style
for col, color in color_map.items():
    # 给lambda添加默认参数c=color，绑定当前迭代的color值
    df_style = df_style.applymap(lambda x, c=color: f'color: {c}' if x > 0 else '', subset=[col])

方法2：用functools.partial绑定参数

通过partial把颜色参数提前绑定到样式函数上，避免闭包陷阱：

from functools import partial

def style_cell(x, color):
    return f'color: {color}' if x > 0 else ''

color_map = {'A': 'red', 'B': 'blue', ..., 'Z': 'green'}
df_style = df.style
for col, color in color_map.items():
    df_style = df_style.applymap(partial(style_cell, color=color), subset=[col])

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