How to display the matrix value and colormap in Matplotlib?

To display the matrix value and colormap in Matplotlib, we can take the following steps −

• Set the figure size and adjust the padding between and around the subplots.
• Create a figure and a set of subplots.
• Initialize max and min values for matrix.
• Plot the values of a 2D matrix or array as color-coded image.
• Iterate each cell of the color-code image and place value at the center.
• To display the figure, use show() method.

Example

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

fig, ax = plt.subplots()
min_val, max_val = 0, 5
matrix = np.random.randint(0, 5, size=(max_val, max_val))
ax.matshow(matrix, cmap='ocean')

for i in range(max_val):
   for j in range(max_val):
      c = matrix[j, i]
      ax.text(i, j, str(c), va='center', ha='center')

plt.show()

Output