如果年龄等于数字，Python Pandas 如何更改背景颜色？

Question

如果满足值等于33的条件，如何更改Age列的颜色？

我的代码：

import pandas as pd

df = pd.DataFrame.from_dict(
    {
        "Nombre": ["Mike", "Jordan", "John"],
        "Age": [33, 45, 20],
        "Lugar": ["Arg", "Pol", "Ind"]
    }
)
def _color_red_or_green(val):
    color = 'red' if val != 33 else 'green'
    return 'color: %s' % color

df.style.applymap(_color_red_or_green)

print(df)

Answer 1

如果您需要在输入 33 时将颜色变为红色，则需要进行此更改。

val == 33

代替

val != 33

这是以防万一的最终代码：

import pandas as pd

df = pd.DataFrame.from_dict(
    {
        "Nombre": ["Mike", "Jordan", "John"],
        "Age": [33, 45, 20],
        "Lugar": ["Arg", "Pol", "Ind"]
    }
)
def _color_red_or_green(val):
    color = 'red' if val == 33 else 'green'
    return 'color: %s' % color

df.style.applymap(_color_red_or_green)

print(_color_red_or_green(33))

Answer 2

如果您只想更改 Age 列的颜色，请尝试以下操作：

# only style the Age column
df.style.applymap(_color_red_or_green, subset=['Age'])

根据您要着色的内容，可以选择以下两个选项之一：

# change color of values
def _color_red_or_green(val):
    color = 'red' if val == 33 else 'green'
    return 'color: %s' % color

# change color of cells
def _color_red_or_green(val):
    color = 'red' if val == 33 else 'green'
    return 'background: %s' % color

Answer 3

使用已经存在的函数，而不是使用 applymap，让它变得简单方便：

(df.style
 .set_properties(subset=["Age"], background="lightgreen")
 .highlight_between(subset=["Age"], color="pink", left=33, right=33)
)

（如果需要，您当然可以使用 其他颜色- 作为背景，我选择了较浅的红色和绿色变体。）