在子图之外放置多个颜色条 (matplotlib)

Question

我有一个包含多个子图行的图形，它们都共享一个 x 轴。 有些行需要颜色条，但其他行不需要。 如果我只使用颜色条功能，子图将会错位。 如何将颜色条放置在子图之外，以便所有行仍然对齐？

Answer 1

我做了一个可能有帮助的函数：

import numpy as np
from matplotlib import pyplot as plt

#function to add colorbar for imshow data and axis
def add_colorbar_outside(im,ax):
    fig = ax.get_figure()
    bbox = ax.get_position() #bbox contains the [x0 (left), y0 (bottom), x1 (right), y1 (top)] of the axis.
    width = 0.01
    eps = 0.01 #margin between plot and colorbar
    # [left most position, bottom position, width, height] of color bar.
    cax = fig.add_axes([bbox.x1 + eps, bbox.y0, width, bbox.height])
    cbar = fig.colorbar(im, cax=cax)

#Example code:
x = np.random.random((10, 100))
fig, axes = plt.subplots(5,1, sharex = True)
im = axes[0].imshow(x, cmap = "Reds", aspect="auto", origin="lower")
add_colorbar_outside(im, axes[0])
im2 = axes[2].imshow(x, cmap = "coolwarm", aspect="auto", origin="lower")
add_colorbar_outside(im2, axes[2])
plt.show()

Answer 2

sharex=True 和 layout='constrained' 参数（如 @drmuelr 指出的）应该足以解决您的问题。
即 fig,axes = plt.subplots(3,1, sharex = True,layout='constrained') 在您的情况下。

Matplotlib 现在有关于如何排列颜色条的良好文档： https:// matplotlib.org/stable/users/explain/axes/colorbar_placement.html