我有以下數據框comp： time first_max name second_max name.1 Perceived OoM.1 Perceived OoM.20 0.000000 18 shoulder_center 9 right_hip shoulder_center shoulder_center1 0.010000 18 shoulder_center 9 right_hip shoulder_center shoulder_center2 0.020000 18 right_hip 9 right_hip shoulder_center right_hip3 0.030000 18 shoulder_center 9 right_hip shoulder_center right_hip我有這個功能，可以根據是否突出顯示整行name == Perceived OoM.1：def highlight_col(x): df = x.copy() mask = df['name'] == df['Perceived OoM.1'] df.loc[mask, :] = 'background-color: yellow' df.loc[~mask,:] = 'background-color: ""' return dfcomp.style.apply(highlight_col, axis=None)但是，我想找出一種方法來為整行著色另一種顏色 if name == Perceived OoM.2。基本上，我希望該行為黃色 ifname == Perceived OoM.1否則該行為藍色 if name == Perceived OoM.2。但我似乎無法將該條件應用到我的函數中。有什么幫助嗎？
2 回答
絕地無雙
TA貢獻1946條經驗 獲得超4個贊
創建另一個掩碼并以相同的方式傳遞，同樣對于默認空值使用DataFrame構造函數：
def highlight_col(x):
df = pd.DataFrame('', index=x.index, columns=x.columns)
mask1 = x['name'] == x['Perceived OoM.1']
mask2 = x['name'] == x['Perceived OoM.2']
df.loc[mask1, :] = 'background-color: yellow'
df.loc[mask2, :] = 'background-color: blue'
return df
拉莫斯之舞
TA貢獻1820條經驗 獲得超10個贊
另一種方法是定義一個函數，這樣你就可以在行上應用：
def highlight(x):
color = 'background-color:yellow' if x['name']==x['Perceived OoM.1']\
else 'background-color: green' if x['name']==x['Perceived OoM.2']\
else ''
return [color]*len(x)
df.style.apply(highlight, axis=1)
輸出：
添加回答
舉報
0/150
提交
取消