如何在Matplotlib中通过按钮退出while循环

尝试通过按钮退出while循环，预期逻辑可行，但首次点击按钮启动循环后，多次点击会导致程序崩溃。怀疑问题出在bprev.on_clicked(turn_off)代码，要么未按预期工作，要么被“跳过”。相关代码如下：

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.widgets import Button
import time

onoff = True
fig, ax = plt.subplots()
axprev = fig.add_axes([0.7, 0.05, 0.1, 0.075])
bprev = Button(axprev, 'On/Off')

def turn_on(val):
    global onoff
    onoff = True
    main()

def main(): 
    global onoff

    def turn_off(val):
        global onoff       
        onoff = False       
    
    while onoff: #main loop
        bprev.on_clicked(turn_off) 
        print("It works ? Probably not.")
        time.sleep(0.4)        

bprev.on_clicked(turn_on) 
plt.show()

问题原因 #

• 重复绑定回调：每次进入while循环都执行bprev.on_clicked(turn_off)，导致按钮被绑定了成百上千个相同的回调函数，多次点击后会同时触发所有回调，直接导致程序崩溃。
• 阻塞事件循环：time.sleep()会完全阻塞Matplotlib的事件处理线程，按钮点击事件无法被及时响应，onoff变量的更新无法同步到循环的判断条件中，导致循环无法退出。

修复后的代码 #

import matplotlib.pyplot as plt
from matplotlib.widgets import Button

onoff = False
fig, ax = plt.subplots()
axprev = fig.add_axes([0.7, 0.05, 0.1, 0.075])
bprev = Button(axprev, 'Start')

def toggle_loop(val):
    global onoff, bprev
    onoff = not onoff
    # 更新按钮文本，直观显示状态
    bprev.label.set_text('Stop' if onoff else 'Start')

def main():
    global onoff
    while onoff:
        print("Loop is running...")
        # 使用plt.pause替代time.sleep，不阻塞事件循环
        plt.pause(0.4)

# 只绑定一次按钮回调
bprev.on_clicked(toggle_loop)
plt.show()

关键修改说明 #

1. 统一回调逻辑：用单个toggle_loop函数实现启停切换，避免分开绑定多个回调，同时更新按钮文本提示当前状态。
2. 避免重复绑定：按钮回调只在初始化时绑定一次，不会在循环中重复添加。
3. 替换阻塞延迟：用plt.pause(0.4)替代time.sleep(0.4)，既能实现延迟效果，又能让Matplotlib处理按钮点击等事件。
4. 初始状态修正：将onoff初始值设为False，程序启动后不会自动进入循环，需点击按钮启动。

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