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显示范围，更好比较