pandas color cell based on value of other column

Question

I would like to color in red cells of a DataFrame on one column, based on the value of another column.

Here is an example:

df = pd.DataFrame([
  { 'color_A_in_red': True , 'A': 1 },
  { 'color_A_in_red': False , 'A': 2 },
  { 'color_A_in_red': True , 'A': 2 },
])

should give:

I know how to color a cell of a df in red but only based on the value of this cell, not the value of another cell:

df_style = df.style
df_style.applymap(func=lambda x: 'background-color: red' if x == 2 else None, subset=['A'])
df_style

Is there a way to color cells of a DataFrame based on the value of another column ?

jezrael · Accepted Answer · 2022-03-16 11:47:15Z

5

Use custom function for DataFrame of styles is most flexible solution here:

def highlight(x):
    c = f"background-color:red" 
    #condition
    m = x["color_A_in_red"]
    # DataFrame of styles
    df1 = pd.DataFrame('', index=x.index, columns=x.columns)
    # set columns by condition
    df1.loc[m, 'A'] = c
    return df1


df.style.apply(highlight, axis=None)

edited Mar 16, 2022 at 11:47

answered Mar 16, 2022 at 11:38

jezrael

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

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

3 Comments

Michael Over a year ago

How to remove color_A_in_red column after adding style attribute when exporting to excel for example. For me it gives error message.

jezrael Over a year ago

@Michael - do you need hide ?

Michael Over a year ago

I have used hide option but it doesn't work for me for some reason.

Collectives™ on Stack Overflow

pandas color cell based on value of other column

1 Answer 1

3 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related