matplotlib：如何返回matplotlib對象然后將其繪制為子圖？

Question

我檢查了這個Matplotlib返回一個繪圖對象，但它確實不適合我的問題。

我想做的是：

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()