在重新创建循环中关闭 window 后的 Matplotlib 图 key_event 连接

Matplotlib figure key_event connection after closing window in recreating loop

我有以下运行良好的极简代码:一个连续的 while 循环不断绘制我的数据,如果我按下转义键,绘制就会停止。现在,如果关闭 matplotlib-window,由于 plt.pause 命令会出现一个新的,但现在 key_event 不再附加。有没有办法保持新出现的 windows 和 key_event 的联系?

代码:

import matplotlib.pyplot as plt
import numpy as np

keep_ploting = True


def action():
    def key_event(event):
        if event.key == 'escape':
            global keep_ploting
            keep_ploting = False

    fig = plt.figure()
    while keep_ploting:
        plt.clf()
        x = np.linspace(1, 10, 100)
        y = np.random.weibull(2,100)

        plt.plot(x, y)
        plt.pause(1e-1)
        fig.canvas.mpl_connect('key_press_event', key_event)

action()

当您关闭 window 时,它会创建新的 figure,您应该使用 gcf()(获取当前数字)将 event 分配给新的 figure

while keep_ploting:
    plt.clf()
    x = np.linspace(1, 10, 100)
    y = np.random.weibull(2,100)

    plt.plot(x, y)
    plt.pause(1e-1)
    
    fig = plt.gcf()  # get current figure
    fig.canvas.mpl_connect('key_press_event', key_event)