Styling dataframe if contain string in all columns with Pandas

Question

The objective is to highlight a cell if it contain * or **.

I have the impression this can be achieved as below

import numpy as np
import pandas as pd

df = pd.DataFrame ( {'Data': ['foo', '*', 'bar'],
                     'myda': ['**', '*', 'wer']} )


def highlight_ (s, props=''):
    return np.where ( s.str.contains ( "*" ), props, '' )


df.apply ( highlight_, props='background-color:yellow', axis=1 )

However, the compiler return an re.error.

re.error: nothing to repeat at position 0

May I know how to properly implement this?

also i think you want df.style.apply(... instead of just df.apply(... — tdy
– tdy, Commented Sep 4, 2021 at 8:09

tdy · Accepted Answer · 2021-09-04 08:34:11Z

2

The asterisk needs to be escaped: '\*'
The highlighting should be applied to df.style instead of just df

def highlight_(s, props=''):
    return np.where(s.str.contains('\*'), props, '')

df.style.apply(highlight_, props='background-color:yellow; color:black;')

To differentiate * and **, try np.select:

def highlight_(s):
    return np.select(
        condlist=[s.str.contains('\*\*'), s.str.contains('\*')],
        choicelist=['background-color:green', 'background-color:purple'],
        default='')

df.style.apply(highlight_)

edited Sep 4, 2021 at 8:34

answered Sep 4, 2021 at 8:14

tdy

42k42 gold badges124 silver badges125 bronze badges

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

2 Comments

rpb Over a year ago

Thanks for the suggestion @tdy. Bit greedy here, but can you suggest how can we have diffrent color for separate case of single and double *

tdy Over a year ago

@balandongiv see if the update works for you

Collectives™ on Stack Overflow

Styling dataframe if contain string in all columns with Pandas

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related