将单个颜色条添加到多个健康子图

Question

我想向包含多个plt.colorbar图的图形添加自定义healpy。我发现很多关于如何在多个axes对象的通常情况下执行此操作的帖子，但是healpy使其很难。

到目前为止，我拥有以下MWE：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl
import healpy as hp

rows, cols = 8, 8
nplots = rows * cols
npix = 48
data = np.random.uniform(size=(nplots, npix))
fig = plt.figure()
for i in range(len(data)):
    hp.mollview(data[i, :], title='', cbar=False, fig=fig,
                sub=(rows, cols, i+1), margins=(0, 0, 0, 0),
                min=data.min(), max=data.max())

fig, ax = plt.gcf(), plt.gca()
image = ax.get_images()[0]
norm =  mpl.colors.Normalize(vmin=data.min(), vmax=data.max())

from mpl_toolkits.axes_grid1 import make_axes_locatable
divider = make_axes_locatable(ax)
cax = divider.new_vertical(size="5%", pad=0.7, pack_start=True)
fig.add_axes(cax)
fig.colorbar(image, cax=cax, norm=norm, orientation='horizontal',
             label='colorbar')
plt.show()

See the erroneous plot here

如链接的图像所示，我最后将一个colorbar附加到最后一个ax，而不是整个fig。我想在colorbar的底部（或右侧）使用一个简单的fig，其范围如上所述通过Normalize指定。同样，事实是我正在使用healpy来产生该图，该图排除了通常的解决方案，至少据我所知。

Answer 1

我没有安装healpy，但是可能这个库只是创建自己的轴。下面的代码模拟了这种情况。您可以从fig.axes获取轴。 与this tutorial中一样，可以通过给出所有“轴”的列表来放置默认颜色条（ax或多或少是子图的matplotlib名称）：plt.colorbar(im, ax=fig.axes)。如果颜色条太大，则它具有一个shrink=0.6参数。

from matplotlib import pyplot as plt
import numpy as np

fig = plt.figure(figsize=(20, 6))

nrows = 4
ncols = 6
for i in range(1, nrows + 1):
    for j in range(1, ncols + 1):
        plt.subplot(nrows, ncols, (i - 1) * ncols + j, projection="mollweide")
        arr = np.random.rand(18, 36)
        Lon, Lat = np.meshgrid(np.linspace(-np.pi, np.pi, 36 + 1), np.linspace(-np.pi / 2., np.pi / 2., 18 + 1))
        plt.pcolormesh(Lon, Lat, arr, cmap=plt.cm.hot)
im = fig.axes[0].collections[0] # or fig.axes[0].get_images()[0] when created as image
plt.colorbar(im, ax=fig.axes)

plt.show()

请注意，在您的代码中，fig已指向当前图形，因此fig = plt.gcf()不必要。 ax = plt.gca()指示上次激活的ax。在示例图的情况下，这似乎是右下角的图。因此，这有助于查找示例图像，但不能将颜色栏放置在所有子图旁边。

如果您需要对颜色栏位置的更多控制，还可以采用this post中的方法：

fig.subplots_adjust(right=0.85)
cbar_ax = fig.add_axes([0.90, 0.15, 0.03, 0.7])
fig.colorbar(im, cax=cbar_ax)