python - 如何在 Matplotlib 中设置颜色条范围？

Question

我使用 Matplotlib 中的 quiver 函数绘制了不同的矢量场。

plt.figure()

x, y = numpy.mgrid[-x_dim/2:x_dim/2:x_steps*1j, -y_dim/2:y_dim/2:y_steps*1j]

m = numpy.sqrt(numpy.power(vector_field_x, 2) + numpy.power(vector_field_y, 2))

fig = plt.quiver(x, y, vector_field_x, vector_field_y, m)
cb = plt.colorbar(fig)

# Add some margin
l, r, b, t = plt.axis()
dx, dy = r-l, t-b
plt.axis([l-0.1*dx, r+0.1*dx, b-0.1*dy, t+0.1*dy])

plt.savefig(file_path + '.png', dpi=Vc.dpi)
plt.close()

它们都是相关的，我需要能够比较它们。

如果我只是绘制它们，我会得到这样的图表:

可以看出，由于它们的最大幅度不同，所以它们以不同的尺度(使用不同的边界)表示。因此，我的第一种方法是将它们一起归一化(即考虑所有矢量场的最大幅度)。

在那之后，我只是绘制了向量，并且比例发生了一点变化。但是，边界仍会根据每个地 block 的最小值和最大值进行调整。

然后，我尝试添加:

cb.set_clim(vmin=0, vmax=1)

使用这段代码，我得到了这样的彩条:

也就是说，我的颜色条范围固定在 0 到 1 之间，所以橙色在所有图中都是 0.8。这就是我想要的。但是，我想绘制“完整”颜色条，表示从 0 到 1。

是否可以在不创建自定义颜色条的情况下实现？

谢谢。

Answer 1

quiver 有一个可选参数 clim:

fig = plt.quiver(x, y, vector_field_x, vector_field_y, m, clim=[-2,2])

这应该有效。 :)