如何在python dataframe中为选中的列着色？

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [6, 7, 8, 9, 10],
        'C': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)

# 定义一个函数来为选中的列着色
def color_selected_columns(value):
    if value in df['A'].values:
        return 'background-color: yellow'
    elif value in df['B'].values:
        return 'background-color: green'
    else:
        return ''

# 创建一个Styler对象，并应用颜色函数
styled_df = df.style.applymap(color_selected_columns)

# 显示着色后的DataFrame
styled_df