python - Matplotlib，关闭循环的按钮，python

Question

我有一个允许用户单击数据点的图。然后它生成一个连续的循环，应该通过单击关闭按钮来关闭该循环。下面的示例将每 1 秒触发一条“Still Looping”打印消息。然后我创建了一个按钮，希望通过将 LoopBool 更改为 True 来关闭循环。但它不起作用，因为一旦循环开始，我就无法与图形窗口交互。我该如何解决这个问题？非常感谢

import numpy as np
from matplotlib import pyplot as plt
import time
import matplotlib.widgets as widgets
fig, ax = plt.subplots()
# it is not always column 0 and 1
sctPlot = ax.scatter([-0.3,0,0.3], [0.3,0.3,0.3],  c="blue", picker = 2, s=[50]*3)
fig.subplots_adjust(bottom=0.3, left=0.1)
plt.grid(False)
plt.axis([-0.6, 0.6, -0.6, 0.6])

loopBool = True

def closeLooping(event):
    global loopBool
    loopBool = False

def looping(event):
    global loopBool
    while (loopBool == True):
        print "Still Looping!"
        time.sleep(1)
    print "Stop!!"

axCloseButton = plt.axes([0.1, 0.15, 0.2, 0.06])
bClose = Button(axCloseButton, "Close", color = axcolor, hovercolor = '0.975')
bClose.on_clicked(closeLooping)


fig.canvas.mpl_connect('pick_event', looping)

Answer 1

在我看来，陷入无限循环的进程在某种意义上与 GUI 不兼容。 GUI 本身执行一个循环，检查并响应发生的事件。最好的解决方案可能是通过将这部分代码转换为基于事件的代码来消除无限循环。

但是，我也找到了解决您实际问题的方法。我对所涉及的编程结构不是很熟悉，所以我无法告诉您这个解决方案有多高效或优雅。重点是在单独的线程中运行无限循环，从而防止主 python 进程陷入循环中。这将使 GUI 保持响应。但是，如果您想在程序运行期间中断程序，这可能会导致问题。

代码:

import time
import threading #this is new
import numpy as np
from matplotlib import pyplot as plt
import matplotlib.widgets as widgets

fig, ax = plt.subplots()
# it is not always column 0 and 1
sctPlot = ax.scatter([-0.3,0,0.3], [0.3,0.3,0.3],  c="blue", picker = 2, s=[50]*3)
fig.subplots_adjust(bottom=0.3, left=0.1)
plt.grid(False)
plt.axis([-0.6, 0.6, -0.6, 0.6])

loopBool = True

def closeLooping(event):
    global loopBool
    loopBool = False

def looping(event):
    global loopBool
    while (loopBool == True):
        print("Still Looping!")
        time.sleep(1)
    print("Stop!!")

def looping_pre(event): #this is new
    thread = threading.Thread(target=looping, args=(event,))
    #thread.daemon = True   #might or might not be needed
    thread.start()

axCloseButton = plt.axes([0.1, 0.15, 0.2, 0.06])
bClose = widgets.Button(axCloseButton, "Close", hovercolor = '0.975')
bClose.on_clicked(closeLooping)
plt.show() #this is new

fig.canvas.mpl_connect('pick_event', looping_pre) #this is changed

请注意，我更改了一些内容，因为您的确切代码似乎无法正常运行。我从 Button 调用中删除了 color=axcolor ；我在事件连接之前添加了一个 plt.show() ，否则我不会出现图形窗口(既不是通过 ipython ，也不是使用 python)。

相关的补充是threading模块和looping_pre前端，它将looping函数作为单独的Thread调用。因此，'pick_event' 不会调用 looping，而是调用 looping_pre。

此代码将(在 ipython 中运行时)显示图形窗口，在单击数据时开始循环，然后在单击按钮时停止循环。然而，当我按下 ctrl+c 时，循环继续进行，因为它是一个单独的线程。我只能通过使用重置来杀死它，从而删除全局loopBool的值。指定Thread是否应deamon化的注释行应该影响此行为(我的意思是这对我来说似乎是合乎逻辑的)，但我没有看到任何效果。