How to plot a 2D matrix in Python with colorbar Matplotlib?

To plot a 2D matrix in Python with colorbar, we can use numpy to create a 2D array matrix and use that matrix in the imshow() method.

Steps

• Create data2D using numpy.

• Use imshow() method to display data as an image, i.e., on a 2D regular raster.

• Create a colorbar for a ScalarMappable instance *mappable* using colorbar() method and imshow() scalar mappable image.

• To display the figure, use show() method.

Example

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
data2D = np.random.random((50, 50))
im = plt.imshow(data2D, cmap="copper_r")
plt.colorbar(im)
plt.show()

Output