python-3.x - 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的文档说:

Add an axes at position rect [left, bottom, width, height] where all quantities are in fractions of figure width and 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])