python - matplotlib:如何返回一个 matplotlib 对象然后作为子图绘制？

Question

我检查了这个 SO Matplotlib returning a plot object但它并不真正适合我的问题。

我想做的是:

def func1():
   fig1 =  plt.plot (np.arange(0.0, 5.0, 0.1))
   return fig1

def func2()
   return plt.plot (np.arange(0.0, 5.0, 0.02))


fig1 = func1()
fig2 = func2()
plt.figure()
plt.add_subplot(fig1)
plt.add_subplot(fig2)
plt.show()

上面的代码只是一个主要思想。你能建议我怎么做吗？

谢谢

Answer 1

这个想法是让你的函数绘制到一个轴上。要么将此坐标轴作为函数的参数提供，要么让它采用当前坐标轴。

import matplotlib.pyplot as plt
import numpy as np

def func1(ax=None):
    ax = ax or plt.gca()
    line, = ax.plot (np.arange(0.0, 5.0, 0.1))
    return line

def func2(ax=None):
    ax = ax or plt.gca()
    line, = ax.plot (np.arange(0.0, 5.0, 0.02))
    return line


fig, (ax1,ax2) = plt.subplots(ncols=2)
func1(ax1)
func2(ax2)

plt.show()