可以使用pandas和style属性来比较数据框内容并更改列的颜色。以下是示例代码:
import pandas as pd
# 创建数据框
df = pd.DataFrame({'name': ['Alice', 'Bob', 'Charlie', 'David'],
'age': [25, 30, 35, 40],
'income': [50000, 60000, 70000, 80000]})
# 创建条件
condition = df['income'] > 60000
# 格式化颜色
def color_negative_red(val):
color = 'red' if val else 'black'
return f'color: {color}'
# 应用样式
styled_df = df.style.applymap(color_negative_red, subset=pd.IndexSlice[condition, 'income'])
# 显示样式
styled_df
这个示例代码比较数据框df
的income
列是否大于60000,并将不满足条件的单元格设置为红色。可以使用applymap()
方法来应用自定义格式化函数,并使用subset()
方法选择要应用格式化的列。最后,使用style()
属性将样式应用于数据框,然后styled_df
将显示带有颜色格式化的数据框。