CODEWITHSUNDEEP
pythonpandas
oyed
oyed

Reputation: 421

How to color whole rows in Python Pandas based on value?

So I figured out how to color singular cells and coloumns using the Style module in pandas. How ever, I want to color entire rows based on values in a singler cell in that row.

For instance, if a cell C in coloumn A has the value 23, color C's row in yellow.

How do I that?

Upvotes: 3

Views: 9777

Answers (1)

jezrael
jezrael

Reputation: 862431

Use:

df =  pd.DataFrame({'A':[23,25,10], 'B':[7,8,3], 'C':[8,3,1]})
print (df)
    A  B  C
0  23  7  8
1  25  8  3
2  10  3  1

def highlight_col(x):
    #copy df to new - original data are not changed
    df = x.copy()
    #set by condition
    mask = df['A'] == 23
    df.loc[mask, :] = 'background-color: yellow'
    df.loc[~mask,:] = 'background-color: ""'
    return df    

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

picture

Upvotes: 7

Related Questions