python - 如何在同一窗口(图)中绘制来自不同函数的 2 个子图？

Question

出于特定原因，我有两个函数，每个函数都在两个不同的窗口中创建一个图。是否可以在一个窗口中统一这两个图，而不统一功能？谢谢!

编辑:我有 2 个涉及的函数和一个数据库:file1.py 中的函数 1 绘制了一个二维线图:

plt.figure("TEST12") 
ax=plt.subplot(111)
ax.plot(array[:,10])

在 file2.py 中还有我的另一个函数，它绘制了一个填充轮廓:

plt.figure("TEST13")
ax = plt.subplot(111)
ax.contourf(x,y,data)
plt.gca().set_aspect('equal')

如果我像往常一样使用 plt.show，结果是 2 个不同的窗口。

Answer 1

重构您的函数，将要绘制的 Axes 对象作为参数:

def fun1(ax):
    ax.plot(range(5))

def fun2(ax):
    ax.plot(range(5)[::-1])


fig, ax = plt.subplots(1, 1)

fun1(ax)
fun2(ax)

plt.draw()