在 matplotlib imshow() 中，如何更改现有 plot 上的颜色轴范围？

Question

在制作 plot 开始时，我可以使用：

fig, ax = plt.subplots() 
h = ax.imshow(X, vmin=mn, vmax=mx)

一旦 plot 启动，我可以改变主意并使vmin和vmax有所不同吗？

Answer 1

您可以使用set_norm()更改图像的标准化并传递Normalize object：

norm = matplotlib.colors.Normalize(vmin=0, vmax=1)
h.set_norm(norm)

Answer 2

这也有诀窍：

h.set_clim([mn, mx])

并重置为“自动”：

h.set_norm(None)