To plot a density map in Python, we can take the following steps −
Create side, x, y, and z using numpy. Numpy linspace helps to create data between two points based on a third number.
Return coordinate matrices from coordinate vectors using side data.
Create exponential data using x and y (Step 2).
Create a pseudocolor plot with a non-regular rectangular grid using pcolormesh() method.
To display the figure, use show() method.
Example
from matplotlib import pyplot as plt, cm, colors import numpy as np plt.rcParams["figure.figsize"] = [7.00, 3.50] plt.rcParams["figure.autolayout"] = True side = np.linspace(-2, 2, 15) X, Y = np.meshgrid(side, side) Z = np.exp(-((X - 1) ** 2 + Y ** 2)) plt.pcolormesh(X, Y, Z, shading='auto') plt.show()