如何在Pandas中按行索引和列名指定单元格着色并保存到Excel？

可以借助Pandas的Styler工具实现按行指定列的单元格着色，步骤如下：

1. 定义着色逻辑函数 #

编写一个函数，接收每行数据和行索引，根据columns_to_color字典标记需要着色的单元格：

import pandas as pd

def highlight_target_cells(row):
    # 初始化所有单元格为默认样式（白色背景）
    style_list = ['background-color: #ffffff'] * len(row)
    # 获取当前行需要着色的列名列表
    target_columns = columns_to_color.get(row.name, [])
    
    for col_name in target_columns:
        # 检查列名是否存在于DataFrame中，避免报错
        if col_name in row.index:
            # 获取列的索引位置，修改对应样式（这里用浅黄背景）
            col_index = row.index.get_loc(col_name)
            style_list[col_index] = 'background-color: #ffff99'
    return style_list

2. 应用样式并导出到Excel #

调用df.style.apply方法按行应用着色函数，再导出到Excel文件：

# 对DataFrame应用自定义着色样式
styled_data = df.style.apply(highlight_target_cells, axis=1)
# 导出到Excel，设置index=False去掉行索引列
styled_data.to_excel('colored_output.xlsx', index=False)

自定义样式说明 #

• 调整颜色：修改background-color的十六进制值即可，比如#ffcccc（浅红）、#ccffcc（浅绿）。
• 扩展样式：可以添加字体颜色、边框等样式，例如'background-color: #ffff99; color: #ff0000; border: 1px solid #000000'。
• 容错处理：函数中加入了列名存在性检查，避免因columns_to_color中列名错误导致程序崩溃。

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