Force Min and Max values of colormap in pandas style

Question

How to force min and max value range when styling a dataframe column ?

Example :

df = pd.DataFrame(columns = ['column'], data = [-1, 0, 1, 2])
df.style.background_gradient(cmap='coolwarm_r')

This produces a column where value 0 is red ; however I would like 0 to be white, i.e. the middle of the 'coolwarm' colormap.

In this example, my understanding of the problem is that the default colormap interval maps my data [-1 : 2].

I would like to set it to [-2 : 2]

Viktor Reichert · Accepted Answer · 2023-03-25 09:30:42Z

1

You can set the vmin and vmax as follows:

df.style.background_gradient(cmap='coolwarm_r', vmin=-2, vmax=2)

answered Mar 25, 2023 at 9:30

Viktor Reichert

513 bronze badges

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

1 Answer 1