使用matplotlib绘图,改变colorbar, 统一colorbar的显示范围,目的是观看两个图片的差别,首先原始代码
import matplotlib.pyplot as plt
import matplotlib
m = 5
plt.figure(figsize=(15,15))
plt.subplot(121)
plt.imshow(data['output'][m].squeeze(), cmap='jet')
plt.colorbar(shrink=0.5)
plt.title('output')
plt.subplot(122)
plt.imshow(data['label'][m].squeeze(), cmap='jet')
plt.colorbar(shrink=0.5)
plt.title('label')
plt.show()
得到的图片展示如下:
由上图看到colorbar数值范围不一致,图片差异明显,肉眼直观较难比较
设置一致的colorbar,便于比较
import matplotlib.pyplot as plt
import matplotlib
m = 5
plt.figure(figsize=(15,15))
norm = matplotlib.colors.Normalize(vmin=0,vmax=0.8) # 设置colorbar显示的最大最小值
plt.subplot(121)
plt.imshow(data['output'][m].squeeze(), cmap='jet',norm=norm)
plt.colorbar(shrink=0.5)
plt.title('output')
plt.subplot(122)
plt.imshow(data['label'][m].squeeze(), cmap='jet',norm=norm)
plt.colorbar(shrink=0.5)
plt.title('label')
plt.show()
出来的图如下:
设vmin=0,vmax=3,出图如下
文章来源:https://www.toymoban.com/news/detail-508361.html
统一colorbar显示范围,更好比较文章来源地址https://www.toymoban.com/news/detail-508361.html
到了这里,关于python 统一两个子图的colorbar显示范围的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!