使用pandas的.style功能处理数据框的显示效果。首先,读取数据框并将两列差异合并到一列中:
import pandas as pd
df = pd.read_csv("data.csv") # 读取数据框
df["diff"] = df["column1"] != df["column2"] # 比较两列差异并合并到一列中
然后,使用.style功能生成新的数据框样式,并设置格式:
def highlight_diff(data, color="yellow"):
'''
高亮差异单元格的函数
'''
attr = 'background-color: {}'.format(color)
return np.where(data, attr, '')
df.style.apply(highlight_diff, axis=None, subset=["diff"]) # 应用样式到数据框中的差异列
执行上述代码后,数据框中的差异单元格将会使用黄色背景高亮显示。