Matplotlibのimshow（）に凡例を追加するにはどうすればよいですか？

Matplotlibのimshow（）に凡例を追加するには、次の手順を実行できます-

• 図のサイズを設定し、サブプロット間およびサブプロットの周囲のパディングを調整します。
• numpyを使用してランダムデータを作成します。
• カラーマップを初期化します。
• サンプルデータから一意のデータポイントを取得します、ステップ2。
• 凡例に配置するために、各色を異なるラベルと色でプロットします。
• ボックス内の右上隅に凡例を配置します。
• 図を表示するには、 show（）を使用します メソッド。

例

import numpy as np
from matplotlib import pyplot as plt, cm

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

data = np.random.rand(3, 3)
cmap = cm.YlOrBr
unique_data = np.unique(data)
i = 0

for entry in unique_data:
   mycolor = cmap(entry * 255 / (max(unique_data) - min(unique_data)))
   plt.plot(0, 0, "-", color=mycolor, label="%d"%i)
   i += 1

plt.imshow(data, cmap=cmap)
plt.legend(loc="upper right", bbox_to_anchor=(1.25, 1.0))

plt.show()

出力