Python：有没有办法抑制函数的绘图但保留回报？

Question

我有一个生成图像的函数，在pyplot图中绘制它，然后返回图像。功能很长，找到确切的绘图位置并不完全合理。有没有人知道一种方法来简单地抑制图，但让函数正常运行所以我可以得到返回的图像？如果有帮助的话，这是在python 3.6中。

Answer 1

您可以对函数的命名空间进行修补，并使用无操作函数替换绘图函数。 给定

'''This is the to-be patched code, living in some module'''

def plot():
    print('Here is a plot: ##====>')

def do_the_plot():
    plot()

然后

'''Our new code, living some place else'''
def newplot():
    print('I wont do it!')

# The globals() will be replaced by the actual module names
globals()['plot'] = globals()['newplot']
do_the_plot()

打印I wont do it!