直接上代码吧
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
是颜色映射,可以在这个链接 找到详细的颜色表