Python：如何使用在函数外部的函数中定义的图形？

Question

我有此代码：

def functionPlot():
    ax = plt.figure().add_subplot(111)
    ax.plot([1,1])
    return ax

if __name__=="__main__":
   ax=functionPlot()     

我想在函数“ functionPlot”中定义要在“ main”函数中使用的图形。我无法做到这一点。我怎样才能做到这一点？我应该从函数返回什么？

Answer 1

也许您是说以下意思：

import matplotlib.pyplot as plt

def functionPlot():
    fig = plt.figure()
    ax = fig.add_subplot(111)
    ax.plot([1,1])
    return fig

if __name__=="__main__":
    fig = functionPlot()
    # do other things with fig