MatplotlibでFigureからカラーバーのインスタンスを取得する方法

概要

MatplotlibでFigureからカラーバーのインスタンスを取得するには、imshow()が返すScalarMappableオブジェクトをcolorbar()メソッドに渡します。これにより、戻り値としてカラーバーのインスタンスを受け取ることができます。

手順

• -1〜1の範囲の値を持つ10×10のランダムな配列データを生成します。

• imshow()メソッドを使用して、データを2次元の正規ラスタ（画像）として表示します。

• imshow()が返すScalarMappableインスタンスをcolorbar()に渡し、カラーバーを作成すると同時にそのインスタンスを取得します。

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

サンプルコード

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.random.randint(-1, 1, (10, 10))
im = plt.imshow(data, interpolation="nearest")
cbar = plt.colorbar(im)
plt.show()

出力結果

上記のコードを実行すると、ランダムに生成されたデータが画像として表示され、その右側に対応するカラーバーが自動的に描画されます。変数cbarにはカラーバーのインスタンスが格納されているため、後からラベルの追加や色の範囲（vmin・vmax）の変更など、さまざまなカスタマイズを行うことが可能です。