Different color for each column in pandas style

Question

I have a dataframe with several columns and a list with colors associated with each column. I want to highlight the non-blank cells in each column with the associated color.

I've tried iterating over the columns in various ways. The closest thing to success was to put a for loop in the styling function and apply it within a for loop. This correctly highlights the last column, but not the rest.

df=pd.DataFrame({'a':[1,2,3,4],'b':['','',1,''],'c':['a','b','c','']})
df_column_colors=['red','blue','green']

def highlight_cells(value):
    if value=='':
        background_color=None
    else:
        for v in range(len(df_column_colors)):
            background_color=str(df_column_colors[v])
    return 'background-color: %s' % background_color
for i in range(len(df.columns)):
     df2=df.style.applymap(highlight_cells,subset=df.columns[i])

A bit of the actual code that I missed in making it generic. I've edited the function in the question and the results are the same. — bcparker21
– bcparker21, Commented Jul 19, 2019 at 17:25

anky · Accepted Answer · 2019-07-19 17:34:03Z

0

you can do this as below:

d= dict(zip(df.columns,['background-color:'+i for i in df_column_colors]))
#{'a': 'background-color:red', 'b': 'background-color:blue', 'c': 'background-color:green'}
def mycolor(x):
    s=pd.DataFrame(d,index=x.index,columns=x.columns)
    df1=x.mask(x.replace('',np.nan).notna(),s)
    return df1
df.style.apply(mycolor,axis=None)

edited Jul 19, 2019 at 17:34

answered Jul 19, 2019 at 17:26

anky

75.3k11 gold badges46 silver badges76 bronze badges

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

1 Comment

bcparker21 Over a year ago

That worked perfectly, and was much cleaner than my attempt. Thanks!

Liam Shalon · Accepted Answer · 2019-07-19 17:58:45Z

0

Try this:

df = pd.DataFrame({'a':[1,2,3,4],'b':['','',1,''],'c':['a','b','c','']})
df_column_colors=['red','blue','green']

def apply_color(cells):
    color = df_column_colors[df.columns.get_loc(cells.name)]
    colors = []
    for cell in cells:
        if cell == '':
            colors.append('')
        else:
            colors.append('background-color: %s' % color)
    return colors

df.style.apply(apply_color)

edited Jul 19, 2019 at 17:58

answered Jul 19, 2019 at 17:37

Liam Shalon

4422 gold badges5 silver badges19 bronze badges

Collectives™ on Stack Overflow

Different color for each column in pandas style

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related