How to plot and work with NaN values in Matplotlib?

MatplotlibPythonData Visualization

To plot and work with NaN values in matplotlib, we can take the following steps −

  • Create data using numpy with some NaN values.

  • Use imshow() method to display data as an image, i.e., on a 2D regular raster, with a colormap and data (from step 1).

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


import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
data = np.array([[1., 1.2, 0.89, np.NAN],
   [1.2, np.NAN, 1.89, 2.09],
   [.78, .67, np.NAN, 1.78],
   [np.NAN, 1.56, 1.89, 2.78]]
plt.imshow(data, cmap="gist_rainbow_r")


Published on 12-May-2021 11:41:33