Пустой Colorbar с использованием basemap.pcolor в ImageGrid

Question

Следующий код создает изображение, я хочу, но colourbar пустой/белый, и не соответствует данным:

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() 

Is есть что-то еще, что мне нужно сделать, чтобы заставить colorbar работать с ImageGrid, или ImageGrid и Basemap просто не хорошо сочетаются?

Answer 1

Насколько я понимаю, это не проблема с базовым картографическим кодом. В вызове ImageGrid() просто устанавливается пустой набор осей для цветовой панели, который можно передать в plt.colorbar() через аргумент cax. По умолчанию ImageGrid() имеет аргумент cbar_set_cax=True, который устанавливает атрибут cax на каждой из подзаговоров. Затем вы можете передать это colorbar():

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

Это будет просто использовать в настоящее время активных осей, чтобы определить диапазон данных Colorbar, так что это будет только в соответствии с окончательным сюжетом. Вам нужно будет использовать комбинации vmin/vmax с pcolormesh(), чтобы обеспечить согласованность по всем участкам.

Answer 2

Matplotlib не может определить, что вы хотите построить на цветной панели. Напишите как

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