3个子图中2个的水平颜色条

Question

有人知道如何优雅地在三个子图中的两个上绘制水平颜色条，在第三个子图上绘制一个额外的水平颜色条。

理想情况下，色条应具有与相应图像轴相同的x尺寸。除了使用matplotlib.gridspec设置整个图像网格外，我没有找到任何好的解决方案。

绘制单个颜色条与mpl_toolkits.axes_grid1.ImageGrid完美匹配，但在尝试在三个轴中的两个轴上绘制颜色条时失败。

Answer 1

我们的想法是使用fig.add_axes添加颜色条所在的轴。 这是一个例子：

import numpy as np
import matplotlib.pyplot as plt

matrix = np.random.random((10, 10, 3))

fig, ax = plt.subplots(1,3, figsize=(12, 3))
plt.subplots_adjust(left=0.05, right=0.85)
for i in range(3):
    im = ax[i].imshow(matrix[:, :, i], interpolation='nearest')
    ax[i].set_aspect("equal")

plt.draw()
p0 = ax[0].get_position().get_points().flatten()
p1 = ax[1].get_position().get_points().flatten()
p2 = ax[2].get_position().get_points().flatten()
ax_cbar = fig.add_axes([p0[0], 0, p1[2]-p0[0], 0.05])
plt.colorbar(im, cax=ax_cbar, orientation='horizontal')

ax_cbar1 = fig.add_axes([p2[0], 0, p2[2]-p2[0], 0.05])
plt.colorbar(im, cax=ax_cbar1, orientation='horizontal')

plt.show()

修改 fig.add_axes的文档说：

  

在位置 rect [左，底部，宽度，添加轴，    height ]所有数量均为数字的分数   宽度和高度。

因此，要将颜色条放在图表的顶部，您只需将bottom参数更改为1.

这

ax_cbar = fig.add_axes([p0[0], 0, p1[2]-p0[0], 0.05])
ax_cbar1 = fig.add_axes([p2[0], 0, p2[2]-p2[0], 0.05])

到

ax_cbar = fig.add_axes([p0[0], 1, p1[2]-p0[0], 0.05])
ax_cbar1 = fig.add_axes([p2[0], 1, p2[2]-p2[0], 0.05])