Highlight pandas dataframe columns based on two different conditions

Question

I have the following dataframe comp:

    time        first_max   name             second_max name.1      Perceived OoM.1 Perceived OoM.2
0   0.000000    18          shoulder_center  9          right_hip   shoulder_center shoulder_center
1   0.010000    18          shoulder_center  9          right_hip   shoulder_center shoulder_center
2   0.020000    18          right_hip        9          right_hip   shoulder_center right_hip
3   0.030000    18          shoulder_center  9          right_hip   shoulder_center right_hip

And I have this function that highlights a whole row based on whether 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 df

comp.style.apply(highlight_col, axis=None)

But, I want to figure out a way to colour a whole row another colour if name == Perceived OoM.2. Basically, I want the row to be yellow if name == Perceived OoM.1 otherwise the row to be blue if name == Perceived OoM.2.

But I can't seem to work that condition into my function.

Any help?

jezrael · Accepted Answer · 2020-06-03 12:27:03Z

3

Create another mask and pass same way, also for default empty values is used DataFrame constructor:

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

edited Jun 3, 2020 at 12:27

answered Jun 3, 2020 at 12:19

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

jezrael Over a year ago

@Oam - Sorry, there was typo for both mask, need compare x like x['name'] == x['Perceived OoM.1'], not df['name'] == df['Perceived OoM.1']

Quang Hoang · Accepted Answer · 2020-06-03 12:30:58Z

1

Another way is to define a function so you can apply on rows:

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)

Output:

answered Jun 3, 2020 at 12:30

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

Collectives™ on Stack Overflow

Highlight pandas dataframe columns based on two different conditions

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related