python - matplotlib - 从一个图到另一个图的重复图？

Question

我对 matplotlib 有点陌生。我想做的是编写将几个图形保存到 eps 文件的代码，然后生成一个合成图形。基本上我想做的是有类似的东西

import matplotlib.pyplot as plt
from matplotlib import gridspec

def my_plot_1():
    fig = plt.figure()
    ...
    return fig

def my_plot_2():
    fig = plt.figure()
    ...
    return fig

def my_combo_plot(fig1,fig2):
    fig = plt.figure()
    gs = gridspec.GridSpec(2,2)
    ax1 = plt.subplot(gs[0,0])
    ax2 = plt.subplot(gs[0,1])
    ax1 COPY fig1
    ax2 COPY fig2
    ...

然后我可以在哪里做类似的事情

my_combo_plot( my_plot_1() , my_plot_2() )

并从前两个函数返回的图中复制所有数据和设置，但我不知道如何使用 matplotlib 完成此操作。

Answer 1

由于 pyplot 有点像状态机，我不确定您所要求的是否可行。相反，我会分解出绘图代码，如下所示:

import matplotlib.pyplot as plt

def my_plot_1(ax=None):
    if ax is None:
        ax = plt.gca()
    ax.plot([1, 2, 3], 'b-')

def my_plot_2(ax=None):
    if ax is None:
        ax = plt.gca()
    ax.plot([3, 2, 1], 'ro')

def my_combo_plot():
    ax1 = plt.subplot(1,2,1)
    ax2 = plt.subplot(1,2,2)
    my_plot_1(ax1)
    my_plot_2(ax2)