Styling Pandas Columns


I am working on a dataframe df. I am trying to set the background color of the cells of the ‘SupTrend’ column. The background should be red when the value in ‘Price’ column is lower than the value in ‘SupTrend’ column, and it should be green otherwise. I tried everything, checked the documentation and I can’t find any explanation or solution for my problem. It seems data inside the function is converted to string and I guess I can’t convert it to float. Does someone knows how to solve this?

    def highlight_suptrend(x, data):
           return f"background-color: red;" if data['Price'].all() < data['SupTrend'].all() else f"background-color: green;", data=df[['Price', 'SupTrend']], subset=['SupTrend'])

Here is the result:

enter image description here

Asked By: Luis Rito Filipe



You want to use style.apply rather than style.applymap, and then use np.where for your function return, selecting red where Price is lower than SupTrend and green otherwise (using:

import pandas as pd
import numpy as np

# some mock data

data = {'Price': np.random.randint(1,11,10),
        'SupTrend': np.random.randint(1,11,10)}

df = pd.DataFrame(data)

def highlight_suptrend(x, data):
    return np.where(data['Price'].lt(data['SupTrend']), 
                    "background-color: red", 
                    "background-color: green"), 
               data=df[['Price', 'SupTrend']], 



Answered By: ouroboros1
Categories: questions Tags: , ,
Answers are sorted by their score. The answer accepted by the question owner as the best is marked with
at the top-right corner.