比较数据框中的两列并使用pandas样式突出显示值

import pandas as pd
import numpy as np

data = {'列1': [1, 2, 3, 4, 5],
        '列2': [2, 4, 6, 8, 10]}
df = pd.DataFrame(data)

def highlight_max(data):
    '''
    为数据框中的最大值添加样式
    '''
    is_max = data == data.max()
    return ['background-color: yellow' if v else '' for v in is_max]

df.style.apply(highlight_max, subset=['列1', '列2'])