直接上代碼吧
import matplotlib.pyplot as plt
image=np.random.rand(32,32)
plt.axis('on') # 刻度尺
plt.imshow(image,cmap='viridis',vmax=0.25,vmin=0.01) # vmax vmin用于指定顔色顯示範圍
plt.colorbar(shrink=0.83) # shrink 用于調整colorbar大小
plt.show()
效果

cmap
是顔色映射,可以在這個連結 找到詳細的顔色表