与轴关联的colorbar（）matplotlib的使用

Question

我想用colorbar（）快速绘制轮廓f（）图。

如果我只有一个地块，我会使用：

plt.contourf(X, Y, Z, 20, cmap='RdGy')
plt.colorbar();

但是，如果我使用轴，应该怎么做？即

fig, ax = plt.subplots(2)

ax[0].contourf(X, Y, Z, 20, cmap='RdBu');
ax[1].contourf(Y, X, Z, 20, cmap='RdBu');

ax[0].colorbar(); #does not work

与第一个示例中的plt.colorbar（）一样，我想要比手动定义分隔器等更快的东西。

Answer 1

colorbar是该方法的一种。它使用一个ScalarMappable作为输入，并可选地使用一个轴来放置彩条。

fig, ax = plt.subplots(2)

im = ax[0].contourf(X, Y, Z, 20, cmap='RdBu')
im2 = ax[1].contourf(Y, X, Z, 20, cmap='RdBu')

fig.colorbar(im, ax=ax[0])

在github上讨论了有关将颜色条用作轴方法的讨论。随时贡献。