0


python 统一两个子图的colorbar显示范围

使用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,出图如下

统一colorbar显示范围,更好比较


本文转载自: https://blog.csdn.net/qq_44850232/article/details/125728012
版权归原作者 曦影y 所有, 如有侵权,请联系我们删除。

“python 统一两个子图的colorbar显示范围”的评论:

还没有评论