Эта комбинация (и значения, близкие к этим), кажется, «магически» работает для меня, чтобы сохранить цветную шкалу, масштабированную до графика, независимо от размера дисплея.
plt.colorbar(im,fraction=0.046, pad=0.04)
Он также не требуют совместного использования оси, которая может получить график из квадрата.