python - 如何通过比较两列来为 Pandas 表着色

Question

例如，我有一个包含两列 A 和 B 的数据框。如果 A > B，我想将该行着色为红色；如果 A <= B，我想将该行涂成绿色。知道我该怎么做吗？谢谢!

Answer 1

来自docs :

def highlight_greater(row):

    if row['A'] > row['B']:
        color = 'red'
    elif row['A'] <= row['B']:
        color = 'green'

    background = ['background-color: {}'.format(color) for _ in row]

    return background

df.style.apply(highlight_greater, axis=1)