Python Pandas colour background cell if number equals

Question

I am currently able to change the background of a cell using the following code:

def my_func_blue(val):
    if val in techniques:
        color = 'green'
        return f'background-color: {color}'
    elif val in Fulltechniques:
        color = 'red'
        return f'background-color: {color}'
s = df1.style.applymap(my_func_blue)
s

I want to be able to add another IF statement into

"if val in techniques"

so if the occurrence is MORE THAN 1 it applies the colour:

DF:

Technique_Name  Technique_ID    SOC Alarm   Occurance
0   Sudo and Sudo Caching   T1548.003   002 1
1   Elevated Execution with Prompt  T1548.004   003 1
13  Cloud Account   T1087.004   015 2
14  Cloud Account   T1087.004   032 2
15  Account Manipulation    T1098   016 1

So only the cell which contains Cloud Account would have the background colour of Green

jezrael · Accepted Answer · 2021-06-09 05:19:01Z

1

If logic is more complicated is possible chain multiple conditions, e.g. here m1 with m2 for create Dataframe of styles and if necessary create excel file:

techniques = ['Cloud Account','Account Manipulation']
Fulltechniques = ['Sudo and Sudo Caching']

def my_func_blue(x): 
   c1 = 'background-color: green'
   c2 = 'background-color: red'
   c = ''
   m1 = x.Technique_Name.str.contains('|'.join(techniques))
   m2 = x.Technique_Name.str.contains('|'.join(Fulltechniques))
   m3 = x.Occurance.gt(1)

   df1 = pd.DataFrame(c, index=x.index, columns=x.columns)
   df1 = df1.mask(m1 & m3, c1).mask(m2, c2)
   return df1

(df.style.apply(my_func_blue,axis=None)
         .to_excel('styled.xlsx', engine='openpyxl', index=False))

edited Jun 9, 2021 at 5:19

answered Jun 8, 2021 at 13:14

jezrael

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

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

2 Comments

TiiTcHY Over a year ago

I've changed it for my variables but it appears to stop after the first match where I would need it to work its way through it all. E.G out of my list above only "Sudo and Sudo Caching" is matching as RED and thats all

jezrael Over a year ago

@TiiTcHY - Can you check now? I change isin for Series.str.contains for match all values of lists joined by | for regex or

Collectives™ on Stack Overflow

Python Pandas colour background cell if number equals

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