Matplotlib中批量迁移Axes并自动估算位置实现垂直堆叠

你的问题核心在于move_axes函数默认复用原Axes的位置，导致新添加的Axes都挤在同一个区域。我们可以修改这个函数，让它根据目标图中已有的Axes数量自动计算垂直位置，实现自动堆叠。

修改后的完整代码 #

import matplotlib.pyplot as plt

def myfunc():
    fig, ax = plt.subplots()
    ax.plot(range(5))
    ax.set_title(f"Subplot {plt.gcf().number}")  # 给每个子图加标题区分
    return fig, ax

def move_axes(ax, original_fig, target_fig, position=None):
    """
    Move Axes object from original_fig to target_fig. Sacrifices original_fig.
    Automatically positions axes vertically if position is None.
    """
    if position is None:
        # 获取目标图当前已有的Axes数量
        current_axes_count = len(target_fig.axes)
        # 计算每个Axes应占的高度（均匀分配垂直空间）
        ax_height = 1.0 / (current_axes_count + 1)
        # 设置新Axes的位置：左、底、宽、高（左0.1，宽0.8留左右边距）
        position = [0.1, current_axes_count * ax_height, 0.8, ax_height]
    
    # 执行迁移操作
    ax.remove()
    ax.figure = target_fig
    target_fig.axes.append(ax)
    target_fig.add_axes(ax)
    ax.set_position(position)
    plt.close(original_fig)
    
    # 自动调整所有已有Axes的位置，确保均匀垂直堆叠
    if position is None:
        for idx, existing_ax in enumerate(target_fig.axes):
            existing_ax.set_position([0.1, idx * ax_height, 0.8, ax_height])
    
    return target_fig

# 调用示例：生成3个子图并迁移到目标图
target_fig = plt.figure(figsize=(8, 6))
for _ in range(3):
    fig, ax = myfunc()
    move_axes(ax, fig, target_fig)

plt.show()

关键改进点 #

1. 自动位置计算：当position为None时，根据目标图当前的Axes数量，计算每个Axes应占的垂直高度，确保新Axes放在已有Axes的下方。
2. 全局位置调整：添加新Axes后，重新调整所有已有Axes的位置，让它们均匀分布在目标图的垂直空间里，避免重叠。
3. 边距优化：固定左右边距为0.1，宽度0.8，让子图布局更美观。

为什么原代码会重叠？ #

你原来的move_axes函数默认使用原Axes的位置（通常是[0.125, 0.11, 0.775, 0.77]，也就是整个画布的中心区域），所以每次添加的新Axes都覆盖在同一个位置，自然会重叠。

另一种简洁思路（可选） #

如果不需要保留迁移Axes的逻辑，也可以先收集所有绘图数据，再在目标图中用subplots直接创建垂直堆叠的子图：

import matplotlib.pyplot as plt

def myfunc():
    return range(5)  # 只返回数据，不创建Axes

target_fig, axes = plt.subplots(3, 1, figsize=(8, 6))
for ax in axes:
    data = myfunc()
    ax.plot(data)
    ax.set_title("Subplot")

plt.tight_layout()
plt.show()

这种方式更直观，但如果myfunc里的绘图逻辑很复杂（比如包含自定义样式、注释等），迁移Axes的方式会更高效。

内容的提问来源于stack exchange，提问作者Daniel