python - 根据条件设置 Pandas 造型

Question

我正在尝试根据条件为数据框的文本或单元格着色。这是我的代码。它有效:

def Highlight_Majors(val):

    color = 'blue' if val == "Austria" else 'black'
    return 'color: %s' % color

s = df.style.applymap(Highlight_Majors)
s

字符串“Austria”现在在数据框中突出显示。如果我需要突出显示多个国家/地区怎么办？

这不起作用:

def Highlight_Majors(val):

    color = 'blue' if val == "Austria"|"Belgium" else 'black'
    return 'color: %s' % color

正确的做法是什么？

Answer 1

使用 in 运算符进行集合成员资格测试:

def Highlight_Majors(val):
    return 'color: %s' % ('blue' if val in {"Austria", "Belgium"} else 'black')