Matplotlib.pyplot.imshow添加colorbar和刻度尺

直接上代码吧

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