Styling pandas based on conditions

Question

I am trying to color the text or cells of data frame based on the condition. This is the code I have. It works:

def Highlight_Majors(val):

    color = 'blue' if val == "Austria" else 'black'
    return 'color: %s' % color

s = df.style.applymap(Highlight_Majors)
s

The string "Austria" now appears highlighted in the dataframe. What if I have more than one countries I need to highlight?

This does not work:

def Highlight_Majors(val):

    color = 'blue' if val == "Austria"|"Belgium" else 'black'
    return 'color: %s' % color

What is the right way to do it?

cs95 · Accepted Answer · 2017-08-26 13:28:48Z

1

Use the in operator with a set membership test:

def Highlight_Majors(val):
    return 'color: %s' % ('blue' if val in {"Austria", "Belgium"} else 'black')

answered Aug 26, 2017 at 13:28

cs95

406k106 gold badges744 silver badges797 bronze badges

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

Comments

Prem · Accepted Answer · 2017-08-26 13:26:18Z

0

How about this?

color = 'blue' if any([val==i for i in ["Austria", "Belgium"]]) else 'black'

answered Aug 26, 2017 at 13:26

Prem

12k1 gold badge21 silver badges34 bronze badges

2 Comments

cs95 Over a year ago

@prashanthmanohar Don't use this. It's really slow. I gave it a -1 because it is a bad way of testing. See my answer for details.

Prem Over a year ago

@coldspeed I assume downvote is meant for 'answer not useful' but in this case this approach could be just another way of achieving the same result. Thanks anyway!

Collectives™ on Stack Overflow

Styling pandas based on conditions

2 Answers 2

Comments

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related