Matplotlibでhist2dプロットにカラーバーを追加する方法

Matplotlibのhist2dプロット（2次元ヒストグラム）にカラーバーを追加するには、colorbar()メソッドの引数にスカラーマッピング可能なオブジェクト（ScalarMappable）を渡します。

実装手順

• NumPyを使ってxとyのデータポイントを作成します。

• subplots()メソッドを使って、図とサブプロットを作成します。

• hist2d()メソッドを使って、2次元ヒストグラムを描画します。

• hist2dから返されるスカラーマッピングオブジェクトをもとに、colorbar()でカラーバーを作成します。

• show()メソッドを使って、図を表示します。

コード例

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()

このコードでは、np.random.randn()で正規分布に従う乱数データを生成し、y方向に+5だけオフセットしています。bins=40でビン数を40に設定し、colors.LogNorm()により対数スケールで色分けすることで、度数のばらつきが大きいデータも見やすく可視化できます。

hist2d()は戻り値としてタプルを返し、その4番目の要素（hh[3]）がQuadMeshオブジェクトです。このオブジェクトをfig.colorbar()に渡すことで、ヒストグラムの色と対応したカラーバーが自動的に生成されます。

出力結果