我有一个属性为display(self)的类对象：

import matplotlib.pyplot as plt

class Obj:
    def display(self) -> None:
        fig = plt.figure()
        sub = fig.add_subplot()
        sub.plot(...)
        plt.show()
 
    def dostuff(self) -> 'stuff':            
        ...
        self.display()
        ...
        self.display()
        ...
        self.display()
        return

我使用这个函数可以更好地直观地了解dostuff(self)属性是如何处理它的任务的。它都按预期工作，当注册self.display()命令时，scrypt暂停执行并绘制内容。但是，要恢复，唯一的方法是手动关闭matplotlib窗口，然后程序使用下一个更改重新打开另一个窗口。有没有一种方法可以实现一个按钮，或者有没有一种更好的方法可以查看接下来的更改，而不必每次都关闭并重新打开一个新窗口？