数据写入excel并填充颜色

import pandas as pd


def style_apply(content, colors, back_ground=''):
    if content != None and content in colors.keys():
        return 'background-color: ' + colors[content]
    return back_ground

def style_color(df, colors):
    return df.style.applymap(style_apply, colors=colors)
    
resultMap = {'A':['test1', 'test2', 'test3'], 'B':['test2', 'test4', 'test5']} #写入数据结构示例


filePath = "eee.xlsx"

compare_pd=pd.DataFrame(resultMap)


writer = pd.ExcelWriter(filePath, engine='openpyxl')#创建数据存放路径

colors = {'test1':'#ff0000', 'test2':'#ffff00'}
style_df = style_color(compare_pd, colors)

style_df.to_excel(writer, header=None, index=False, sheet_name='compare')

writer.save()#文件保存
writer.close()