我有以下數(shù)據(jù)框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我有這個(gè)功能，可以根據(jù)是否突出顯示整行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否則該行為藍(lán)色 if name == Perceived OoM.2。但我似乎無法將該條件應(yīng)用到我的函數(shù)中。有什么幫助嗎？
2 回答
絕地?zé)o雙
TA貢獻(xiàn)1946條經(jīng)驗(yàn) 獲得超4個(gè)贊
創(chuàng)建另一個(gè)掩碼并以相同的方式傳遞，同樣對(duì)于默認(rèn)空值使用DataFrame構(gòu)造函數(shù)：
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貢獻(xiàn)1820條經(jīng)驗(yàn) 獲得超10個(gè)贊
另一種方法是定義一個(gè)函數(shù)，這樣你就可以在行上應(yīng)用：
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)
輸出：
添加回答
舉報(bào)
0/150
提交
取消