Python3：导入依赖于Matplotlib的函数

Question

美好的一天！因此，我尝试在自己的文件中使用经常使用的函数用于组织目的，并希望能够在需要时导入它。

#Plotting fits image function
def plotfits(subsize,title,imgdata,color,cbarlabel):
    plt.subplot(subsize)
    plt.title(titSle)
    plt.imshow(imgdata, cmap=color)
    ibar1 = plt.colorbar()
    ibar1.set_label(cbarlabel)
    plt.gca().invert_yaxis()

然后当我在我想要使用该函数的主文件中尝试from plotfits import plotfits时，我收到此错误。

Traceback (most recent call last):
  File "/home/jorge/python/heliotutorial/tut01_new.py", line 39, in <module>
    plotfits(221,'Intensity',imgdata1,'gray','Continuum Intensity')
  File "/home/jorge/python/heliotutorial/plotfits.py", line 4, in plotfits
    def plotfits(subsize,title,imgdata,color,cbarlabel):
NameError: name 'plt' is not defined

但我已经使用import matplotlib.pyplot as plt导入了pyplot。我想也许我必须再次在函数中导入pyplot才能工作，因为它在函数内定义：

def plotfits(subsize,title,imgdata,color,cbarlabel):
    import matplotlib.pyplot as plt
    ....

但这不起作用，也没有在文件开头的import语句。这可能是我的错误？ （如果在主文件中编写和定义，函数确实有效）

提前感谢您的帮助！