在 ImageGrid 中使用 basemap.pcolor 清空颜色栏

Question

下面的代码生成了我想要的图像，但是颜色条是 blank/white，并且与数据不匹配：

def plot_array(da, ax=None, shift=True):
    """plots an array of lat by lon on a coastline map"""

    m = basemap.Basemap()
    m.drawcoastlines()
    m.pcolormesh(da.lon, y=da.lat, data=da.T, latlon=True)

    return m

# 'monthly mean' is an xarray DataArray

fig = plt.figure(0, (14, 8))
grid = ImageGrid(fig, 111, nrows_ncols=(3, 4), axes_pad=0.3,
                 cbar_mode='single', cbar_location="bottom",)
for i, m in enumerate(months):
    plt.sca(grid[i])
    plot_array(monthly_mean.sel(month=i + 1))
    plt.title(m)
plt.suptitle("{b} - {y} monthly means".format(b=benchmark, y=year))
# plt.colorbar()
plt.tight_layout()

os.makedirs("plots/monthly_means/{y}".format(y=year), exist_ok=True)
plt.savefig("plots/monthly_means/{y}/{b}_{y}.png".format(b=benchmark, y=year))
plt.close()

我还需要做些什么才能让颜色栏与 ImageGrid 一起工作，还是 ImageGrid 和底图不能很好地协同工作？

Answer 1

Matplotlib 无法定义要在颜色条上绘制的内容。写喜欢

...
pcm = m.pcolormesh(da.lon, y=da.lat, data=da.T, latlon=True)
plt.colorbar(pcm)
...

Answer 2

据我了解，这不是底图问题。 ImageGrid() 调用只是为颜色栏设置一组空白轴，可以通过 cax 参数传递给 plt.colorbar()。默认情况下 ImageGrid() 有参数 cbar_set_cax=True，它在每个子图中设置 cax 属性。然后您可以将其传递给 colorbar():

...
plt.colorbar(cax=grid[0].cax)
plt.tight_layout()
...

这只会使用当前活动的轴来确定颜色条数据范围，因此它只会与最终绘图保持一致。您将不得不使用 vmin/vmax 和 pcolormesh() 的组合来确保所有绘图的一致性。

在 ImageGrid 中使用 basemap.pcolor 清空颜色栏

Empty colorbar using basemap.pcolor in an ImageGrid

python

matplotlib

subplot

matplotlib-basemap