Data Structure
Networking
RDBMS
Operating System
Java
MS Excel
iOS
HTML
CSS
Android
Python
C Programming
C++
C#
MongoDB
MySQL
Javascript
PHP
Physics
Chemistry
Biology
Mathematics
English
Economics
Psychology
Social Studies
Fashion Studies
Legal Studies
- Selected Reading
- UPSC IAS Exams Notes
- Developer's Best Practices
- Questions and Answers
- Effective Resume Writing
- HR Interview Questions
- Computer Glossary
- Who is Who
How to create a heat map in Python that ranges from green to red? (Matplotlib)
To create a heatmap in Python that ranges from green to red, we can take the following steps −
- Set the figure size and adjust the padding between and around the subplots.
- Make a dictionary for different colors.
- Create a colormap from linear mapping segments using LinearSegmentedColormap.
- Create a figure and a set of subplots.
- Create random data with 5☓5 dimension.
- Create a pseudocolor plot with a non-regular rectangular grid.
- Create a colorbar for a ScalarMappable instance, *mappable*.
- To display the figure, use show() method.
Example
import matplotlib.pyplot as plt
import matplotlib.colors as colors
import numpy as np
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
cdict = {'red': ((0.0, 0.0, 0.0),
(0.5, 1.0, 1.0),
(1.0, 1.0, 0.7)),
'green': ((0.0, 0.7, 0.7),
(0.5, 1.0, 1.0),
(1.0, 0.0, 0.0)),
'blue': ((0.0, 0.0, 0.0),
(0.5, 1.0, 1.0),
(1.0, 0.0, 0.0))
}
GnRd = colors.LinearSegmentedColormap('GnRd', cdict)
fig, ax = plt.subplots(1)
data = np.random.rand(5, 5)*6.-3.
p = ax.pcolormesh(data, cmap=GnRd, vmin=-5, vmax=5)
fig.colorbar(p, ax=ax)
plt.show()Output
Updated on: 07-Jul-2021
923 Views
Advertisements