How to add a colorbar for a hist2d plot in Matplotlib?

MatplotlibPythonData Visualization

To add a colorbar for hist2d plot, we can pass a scalar mappable object to colorbar() method's argument.

Steps

  • Create x and y data points using numpy.

  • Create a figure and a set of subplots using subplots() method.

  • Make a 2D histogram plot using hist2d() method.

  • Create a colorbar for a hist2d scalar mappable instance.

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

Example

import numpy as np
from matplotlib import pyplot as plt, colors
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
x = np.random.randn(100)
y = np.random.randn(100) + 5
fig, ax = plt.subplots()
hh = ax.hist2d(x, y, bins=40, norm=colors.LogNorm())
fig.colorbar(hh[3], ax=ax)
plt.show()

Output

raja
Published on 11-May-2021 13:00:42
Advertisements