如何使用Pandas对Excel中的单个单元格进行条件格式设置？

问题场景 #

现有如下DataFrame：

import pandas as pd
import numpy as np
df = pd.DataFrame({'a': ['long', 'short', np.nan], 'b': [np.nan, 3, 2]})

需要实现：导出Excel时，当df.a == 'long'时a列单元格设为绿色，df.a == 'short'时设为红色。之前尝试的函数无法生效。

解决方案 #

之前的问题在于返回的样式数组维度和原DataFrame不匹配，Styler.apply(axis=None)要求返回与原DataFrame形状完全一致的样式矩阵。以下是两种可行的实现方式：

方法一：返回完整样式矩阵 #

import pandas as pd
import numpy as np

df = pd.DataFrame({'a': ['long', 'short', np.nan], 'b': [np.nan, 3, 2]})

def bounded_highlights(df):
    # 创建与原DataFrame同形状的空样式容器
    style_matrix = pd.DataFrame('', index=df.index, columns=df.columns)
    # 为a列符合条件的单元格设置样式
    style_matrix.loc[df['a'] == 'long', 'a'] = 'background-color: lime'
    style_matrix.loc[df['a'] == 'short', 'a'] = 'background-color: pink'
    return style_matrix

# 应用样式并导出，需指定openpyxl引擎支持样式写入
df.style.apply(bounded_highlights, axis=None).to_excel('styled_result.xlsx', engine='openpyxl', index=False)

方法二：针对目标列单独处理 #

这种方式更简洁，仅对a列应用样式逻辑：

import pandas as pd
import numpy as np

df = pd.DataFrame({'a': ['long', 'short', np.nan], 'b': [np.nan, 3, 2]})

def highlight_a_cell(value):
    if value == 'long':
        return 'background-color: lime'
    elif value == 'short':
        return 'background-color: pink'
    return ''

# 仅对a列应用样式，导出时指定引擎
df.style.applymap(highlight_a_cell, subset=['a']).to_excel('styled_result.xlsx', engine='openpyxl', index=False)

注意事项 #

• 导出带样式的Excel必须指定engine='openpyxl'，若未安装需先执行pip install openpyxl。
• 样式函数需确保返回每个单元格对应的样式字符串，无样式需求时返回空字符串。

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