Matplotlib实现切换显示按钮：避免重复关闭窗口方案

我有一个包含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时，每次调用display都会弹出新窗口，必须手动关闭才能继续执行后续逻辑。想实现无需关闭窗口，就能查看下一次绘图变化的方案，比如通过切换按钮控制。

方法1：复用窗口+按钮控制（手动切换） #

核心是只创建一次图形和坐标轴，后续更新时清除旧内容、重绘新数据，再通过按钮触发下一次显示，避免重复创建窗口。

修改后的代码示例：

import matplotlib.pyplot as plt

# 切换到交互式后端（部分环境默认已开启，确保窗口不阻塞程序）
plt.switch_backend('Qt5Agg')

class Obj:
    def __init__(self):
        # 初始化时创建唯一的画布、坐标轴和控制按钮
        self.fig, self.sub = plt.subplots()
        self.next_btn = plt.Button(plt.axes([0.8, 0.02, 0.15, 0.075]), '下一次')
        self.next_btn.on_clicked(self._trigger_next)
        self.can_proceed = False  # 标记是否允许继续执行

    def _trigger_next(self, event):
        self.can_proceed = True

    def display(self, data):
        # 清除坐标轴旧内容
        self.sub.clear()
        # 绘制新数据
        self.sub.plot(data)
        # 刷新画布显示
        self.fig.canvas.draw()
        self.fig.canvas.flush_events()
        # 等待按钮点击后继续
        self.can_proceed = False
        while not self.can_proceed:
            plt.pause(0.1)

    def dostuff(self):
        # 模拟处理过程中生成的多组数据
        data_groups = [[1,2,3], [3,1,2], [2,3,1]]
        for data in data_groups:
            # 此处添加你的任务处理逻辑
            self.display(data)
        plt.close()

if __name__ == '__main__':
    obj = Obj()
    obj.dostuff()

方法2：自动间隔刷新（无需手动点击） #

如果不需要手动控制节奏，只是想自动切换显示，可使用plt.pause()实现非阻塞式的定时刷新：

import matplotlib.pyplot as plt

plt.switch_backend('Qt5Agg')

class Obj:
    def __init__(self):
        self.fig, self.sub = plt.subplots()

    def display(self, data):
        self.sub.clear()
        self.sub.plot(data)
        plt.pause(2)  # 显示2秒后自动继续执行后续逻辑

    def dostuff(self):
        data_groups = [[1,2,3], [3,1,2], [2,3,1]]
        for data in data_groups:
            # 此处添加你的任务处理逻辑
            self.display(data)
        plt.close()

if __name__ == '__main__':
    obj = Obj()
    obj.dostuff()

方法3：动画模块实现连续播放（适合连续数据） #

如果数据是连续生成的，可使用Matplotlib的FuncAnimation实现自动播放式的绘图更新：

import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

class Obj:
    def __init__(self):
        self.fig, self.sub = plt.subplots()
        self.data_groups = [[1,2,3], [3,1,2], [2,3,1]]
        self.current_idx = 0

    def _update_plot(self, frame):
        self.sub.clear()
        self.sub.plot(self.data_groups[self.current_idx])
        self.current_idx += 1
        # 数据播放完毕后停止动画
        if self.current_idx >= len(self.data_groups):
            self.anim.event_source.stop()
        return self.sub,

    def dostuff(self):
        # 此处添加你的任务处理逻辑，生成data_groups
        self.anim = FuncAnimation(self.fig, self._update_plot, interval=2000, blit=True)
        plt.show()

if __name__ == '__main__':
    obj = Obj()
    obj.dostuff()

内容的提问来源于stack exchange，提问作者ugo_capeto