Remove white border when using subplot and imshow in Python Matplotlib

To remove white border when using subplot and imshow(), we can take the following steps −

• Set the figure size and adjust the padding between and around the subplots.
• Create random data points using numpy.
• Get the size of the data.
• Set the figure sizes in inches.
• Get the axes instance that contains most of the figure element.
• Turn off the axes.
• Add axes to the figure.
• Display the data as an image, i.e., on a 2D regular raster.
• To display the figure, use show() method.

Example

import numpy as np
import matplotlib.pyplot as plt

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

data = np.random.randint(0, 50, (50, 50))
sizes = np.shape(data)

fig = plt.figure()
fig.set_size_inches(1. * sizes[0] / sizes[1], 1, forward=False)

ax = plt.Axes(fig, [0., 0., 1., 1.])
ax.set_axis_off()

fig.add_axes(ax)
ax.imshow(data)

plt.show()

Output

It will produce the following output