How to plot an image with non-linear Y-axis with Matplotlib using imshow?

To plot an image with non-linear Y-axis with matplotlib using imshow() method, we can take the following steps −

• Set the figure size and adjust the padding between and around the subplots.
• Add a subplot to the current figure.
• Set nonlinear Y-axis ticks.
• Create random data points using numpy.
• Display data as an image, i.e., on a 2D regular raster, with data.
• To display the figure, use show() method.

Example

import matplotlib.pyplot as plt
import numpy as np

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

ax = plt.subplot(111)
ax.yaxis.set_ticks([0, 2, 4, 8])

data = np.random.randn(5, 5)

plt.imshow(data, cmap='copper')

plt.show()

Output