将数字添加到子图

我是 matplotlib 的新手，正在尝试了解如何将数字添加到子图中。

我有三个不同的函数，它们输出一个数字：

def plot_fig_1(vars, args):
    f, ax, put.subplots()
    # do something
    ax.plot(x, y)
    return f, ax


def plot_fig_2(vars, args):
    f, ax, put.subplots()
    # do something
    ax.plot(x, y)
    return f, ax

现在，例如，我想将两个图合并为具有共享 X 轴的单个图。我试过：

f_1, ax_1 = plot_fig_1(...)
f_2, ax_2 = plot_fig_2(...)

new_fig, new_ax = plt.subplots(2,1)
new_ax[0] = f_1
new_ax[1] = f_2

在这里我基本上迷路了。我正在阅读 Matplotlib 手册，但到目前为止还没有运气。

除非您的函数签名必须保持示例中定义的那样，否则在函数外部创建子图并传递适当的函数会更容易Axes每个函数的实例。

def plot_fig_1(vars, args, ax):
    # do something
    ax.plot(x, y)

def plot_fig_2(vars, args, ax):
    # do something
    ax.plot(x, y)

fig, ax = plt.subplots(2, 1, sharex=True)
plot_fig_1(..., ax[0])
plot_fig_2(..., ax[1])

如果您需要创建一个仅包含其中一个子图的图形，您可以使用以下命令来执行此操作：

fig, ax = plt.subplot()
plot_fig_1(..., ax)

或者，如果函数需要独立，请给出ax参数一个默认值并在函数内测试它。

def plot_fig_1(vars, args, ax=None):
    if ax is None:
        fig, ax = plt.subplot()
    # do something
    ax.plot(x, y)