Open In App

Make a gradient color mapping on a specified column in Pandas

Improve
Improve
Like Article
Like
Save
Share
Report

Let us see how to gradient color mapping on specific columns of a Pandas DataFrame. We can do this using the Styler.background_gradient() function of the Styler class.

Syntax : Styler.background_gradient(cmap=’PuBu’, low=0, high=0, axis=0, subset=None)

Parameters :  

cmap : str or colormap (matplotlib colormap)

low, high : float (compress the range by these values.)

axis : int or str (1 or ‘columns’ for columnwise, 0 or ‘index’ for rowwise)

subset : IndexSlice (a valid slice for data to limit the style application to)

Returns :  self

Approach :

  • Import Pandas module
  • Create DataFrame
  • Wisely choose specific column with style.background_gradient() function
  • Display DataFrame

Let’s understand with examples:

Example 1 :

Create a DataFrame and gradient all the columns.

Python3




# importing pandas module
import pandas as pd
 
# Creating pandas DataFrame
df = pd.DataFrame({"A": [1, 2, -3, 4, -5, 6],
                   "B": [3, -5, -6, 7, 3, -2],
                   "C": [-4, 5, 6, -7, 5, 4],
                   "D": [34, 5, 32, -3, -56, -54]})
 
# Displaying the original DataFrame
print("Original Array : ")
print(df)
 
# background color mapping
print("\nDataframe - Gradient color:")
df.style.background_gradient()


Output :

Example 2 :

Create a DataFrame and gradient the specific columns

Python3




# importing pandas module
import pandas as pd
 
# Creating pandas DataFrame
df = pd.DataFrame({"A": [1, 2, -3, 4, -5, 6],
                   "B": [3, -5, -6, 7, 3, -2],
                   "C": [-4, 5, 6, -7, 5, 4],
                   "D": [34, 5, 32, -3, -56, -54]})
 
# Displaying the original DataFrame
print("Original Array : ")
print(df)
 
# background color mapping
print("\nDataframe - Gradient color:")
 
# df.style.background_gradient()
df.style.background_gradient(subset='B')


Output :

If you want to change another column then

Python3




df.style.background_gradient(subset='D')


Output : 

,



Last Updated : 05 Apr, 2022
Like Article
Save Article
Previous
Next
Share your thoughts in the comments
Similar Reads