How to color pandas table by comparing two columns

Question

For example, I have a dataframe with two columns A and B. If A > B, I want to color the row red; if A <= B, I want to color that row green. Any idea how can I do this? Thanks!

I've read it. But all its examples take only one value. I want to know how can I compare two values from the dataframe. — qhan
– qhan, Commented Jun 27, 2017 at 20:57

jack6e · Accepted Answer · 2017-06-27 21:18:34Z

9

From the docs:

def highlight_greater(row):

    if row['A'] > row['B']:
        color = 'red'
    elif row['A'] <= row['B']:
        color = 'green'

    background = ['background-color: {}'.format(color) for _ in row]

    return background

df.style.apply(highlight_greater, axis=1)

edited Jun 27, 2017 at 21:18

answered Jun 27, 2017 at 21:00

jack6e

1,51211 silver badges12 bronze badges

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

2 Comments

qhan Over a year ago

Thank you for your answer! When I tried it, an error occurred: ValueError: Function <function highlight_greater at 0x0000027925CD7620> returned the wrong shape. Result has shape: (417,) Expected shape: (417, 2) 417 is the number of rows in my dataframe and 2 is the number of columns. Do you have any idea why the result does not have a shape (417,2)? Thanks!

jack6e Over a year ago

Sorry, forgot to return a color value for each column in the row. Updated - let me know if that works.

Collectives™ on Stack Overflow

How to color pandas table by comparing two columns

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