如何复用matplotlib函数生成的子图并与其他子图重组？

你现在遇到的问题本质是：每个Axes（子图对象）是绑定在特定Figure（画布）上的，没法直接把一个画布的子图移到另一个画布。所以关键是把绘制子图内容的逻辑和创建画布/子图的逻辑分开，这样就能在任意子图上复用绘制逻辑了。

步骤1：拆分绘制逻辑为独立函数 #

先把原来在genFigures里给四个子图绘图的代码，拆成四个独立的小函数，每个负责绘制对应子图的内容：

def _plot_simpF(self, ax):
    # 这里写你原本给barsimpF绘制内容的代码，比如：
    ax.bar([1,2,3], [4,5,6])
    ax.set_title("Simple F Distribution")
    ax.set_xlabel("Category")
    ax.set_ylabel("Count")
    # 其他样式设置也放这里

def _plot_simpT(self, ax):
    # 对应barsimpT的绘制逻辑
    ax.hist(some_data)
    ax.set_title("Simple T Distribution")
    # ... 其他设置

def _plot_sworF(self, ax):
    # 对应barsworF的绘制逻辑
    ax.plot(x, y)
    ax.set_title("Without Replacement F")
    # ... 其他设置

def _plot_sworT(self, ax):
    # 对应barsworT的绘制逻辑
    ax.scatter(x, y)
    ax.set_title("Without Replacement T")
    # ... 其他设置

步骤2：重构genFigures函数，支持外部传入子图 #

修改原函数，让它可以选择自己创建子图或者使用外部传入的子图，这样既保留原来的功能，又支持复用绘制逻辑：

def genFigures(self, savefile='out/histograms.png', show=0, axes=None):
    # 如果没有传入子图，就创建默认的2x2布局
    if axes is None:
        fig, ((barsimpF, barsimpT), (barsworF, barsworT)) = plt.subplots(
            2, 2, sharey=True, figsize=(12,10)
        )
        axes = (barsimpF, barsimpT, barsworF, barsworT)
    else:
        # 传入的axes需是包含4个子图的元组，顺序对应simpF, simpT, sworF, sworT
        barsimpF, barsimpT, barsworF, barsworT = axes
        fig = barsimpF.figure  # 获取子图所属的画布

    # 调用拆分后的绘制函数，给每个子图填内容
    self._plot_simpF(barsimpF)
    self._plot_simpT(barsimpT)
    self._plot_sworF(barsworF)
    self._plot_sworT(barsworT)

    # 保存和显示逻辑保留
    if savefile:
        plt.savefig(savefile, bbox_inches='tight')
    if show:
        plt.show()
    
    return fig, axes  # 返回画布和子图，方便后续操作

步骤3：任意组合子图的用法 #

现在你可以完全自由地组合不同的子图内容了：

例子1：把两次调用的不同子图组合成新画布 #

比如，把第一次调用的simpF和第二次调用的sworT放到一个2行1列的新画布中：

# 1. 创建新的画布和子图布局
fig_combined, (ax_top, ax_bottom) = plt.subplots(2, 1, figsize=(8, 10))

# 2. 用第一个数据集绘制simpF到ax_top
self._plot_simpF(ax_top)  # 直接调用绘制函数最灵活

# 3. 用第二个数据集绘制sworT到ax_bottom
# 如果sworT的绘制需要依赖genFigures的其他逻辑，也可以这样：
temp_ax1 = plt.subplots()[1]
temp_ax2 = plt.subplots()[1]
temp_ax3 = plt.subplots()[1]
self.genFigures(axes=(temp_ax1, temp_ax2, temp_ax3, ax_bottom), savefile=None, show=0)

# 4. 调整布局并保存/显示
fig_combined.tight_layout()
plt.savefig("out/combined_plot.png")
plt.show()

例子2：复用同一份绘制逻辑到不同布局 #

比如，把四个子图改成1行4列的布局：

fig_new, axes_new = plt.subplots(1, 4, figsize=(16, 6), sharey=True)
# 传入新的子图数组，genFigures会自动在上面绘制内容
self.genFigures(axes=axes_new, savefile="out/1x4_layout.png", show=1)

关键提醒 #

• 不要尝试移动已创建的Axes对象：Matplotlib不支持把一个子图从一个画布移到另一个，所以解耦绘制逻辑是唯一可行的方案。
• 如果不同的genFigures调用对应不同的数据集，可以给绘制函数加参数，比如_plot_simpF(self, ax, data)，这样能更灵活地适配不同数据。

内容的提问来源于stack exchange，提问作者andrea m.