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

Question

我有这个代码：

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

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

我想获取函数 "functionPlot" 中定义的图形以用于 "main" 函数。我做不到。我怎样才能做到这一点？我应该 return 从函数中得到什么？

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

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

Python: How to use a figure defined in a function outside the function?

function

figure

python-3.x