如何使用matplotlib从外部功能生成的添加_subplot图

Question

当前，我有三个函数来创建图形，即get_image_1，get_image_2和get_image_3，如下所示。

def get_image_1():
    fig, ax = plt.subplots(figsize=(8, 6))
    plt.plot(range(10))
    fig.savefig('image_1.png')
    # return fig

def get_image_2():
    fig, ax = plt.subplots(figsize=(8, 6))
    plt.plot([0,5,8,9,3,6])
    fig.savefig('image_2.png')
    # return fig

def get_image_3():
    fig, ax = plt.subplots(figsize=(8, 6))
    plt.plot([100,5,8,9,3,6])
    fig.savefig('image_3.png')
    # return fig

为简单起见，我们只使用Simple plt.plot（）。

上述函数通过

get_image_1()
get_image_2()
get_image_3()

我想显示这三个图，如本图所示。

当前，我必须在本地保存每个图像。例如，在函数get_image_1中，请注意行fig.savefig（'image_1.png'）。

并重新上传保存的图像，然后根据下面的代码将它们组合在一起，

import matplotlib.pyplot as plt
import cv2 as cv
fig = plt.figure(figsize=(9, 10))
for idx,dpath in enumerate(['image_1.png','image_2.png','image_3.png']):
    tt=1
    if idx!=2:
        sub1 = fig.add_subplot(2, 2, idx + 1)
    else:
        sub1 = fig.add_subplot(2, 2, (3,4))

    image2 = cv.imread(dpath)
    sub1.imshow(image2, 'gray')

plt.show()

我想知道如何更有效地进行此活动。这样，要完全跳过保存的并重新启动图像。

Answer 1

根据@bigben的建议。

修改函数以接受axes参数

  def get_image_1(ax):
    ax.plot(range(10))


def get_image_2(ax):
    ax.plot([0,5,8,9,3,6])

def get_image_3(ax):
    ax.plot([100,5,8,9,3,6])

，然后传递axes每个function>函数调用参数

fig = plt.figure(figsize=(9, 10))

ax1 = fig.add_subplot(2, 2, 1)
get_image_1(ax1)

get_image_2(fig.add_subplot(2, 2, 2))

get_image_3(fig.add_subplot(2, 2, (3,4)))

plt.show()