imshowした時にcolorbarを表示する

matplotlib.pyplotで２次元の行列をax.imshowすると画像になります。この時に、colorbarを表示する時には、plt.colorbarではダメです。

colorbarの表示

ax.imshow()した時のオブジェクトが必要になります。imに入れています。

Xは二次元の行列です。値は規格化されていなくてもcmapはいい感じにmaxとminを決めて色を振ってくれます。

fig = plt.figure()
ax = fig.add_subplot(111)
im = ax.imshow(X, cmap="gray", aspect="auto")
fig.colorbar(im)
plt.show()