通过检测按下的键来刷新和关闭绘图

Question

我有一个简单的循环来绘制从文件夹中读取的数据。它永远循环更新情节，我想在按 ESC 时结束程序。到目前为止，我写了

fig = plt.figure()
plt.axes()
while True:
    ... # loop over data and plot
    plt.draw()
    plt.waitforbuttonpress(0)
    plt.cla()

如果我通过单击 X 图标关闭图形，程序将以错误结束。我可以通过做来避免错误

    try:
        plt.waitforbuttonpress(0)
    except:
        break

但我仍然希望能够通过在情节上按 ESC 来终止程序。此外，如果我使用 CTRL+W 关闭绘图，则绘图会重新出现。我尝试添加事件检测，例如

def parse_esc(event):
    if event.key == 'press escape':
        sys.exit(0)
fig.canvas.mpl_connect('key_press_event', parse_esc)

但它不会检测到 ESC。我尝试使用close_event而不是key_press_event但sys.exit(0)给出了以下错误

    while executing
"140506996271368filter_destroy 836 ?? ?? ?? ?? ?? ?? ?? ?? ?? ?? 0 ?? ?? .140506996230464 17 ?? ?? ??"
    invoked from within
"if {"[140506996271368filter_destroy 836 ?? ?? ?? ?? ?? ?? ?? ?? ?? ?? 0 ?? ?? .140506996230464 17 ?? ?? ??]" == "break"} break"
    (command bound to event)

我还想删除循环并仅在检测到 R 时才刷新情节，但这并不那么重要。

任何帮助不胜感激，谢谢。

Answer 1

如果有人需要做类似的事情，这就是我所做的

folder = ...
def update():
    plt.cla()
    for f in os.listdir(folder):
        if f.endswith(".dat"):
            data = ... 
            plt.plot(data)
    plt.draw()
    print('refreshed')
def handle(event):
    if event.key == 'r':
        update()
    if event.key == 'escape':
        sys.exit(0)
fig = plt.figure()
plt.axes()
picsize = fig.get_size_inches() / 1.3
fig.set_size_inches(picsize)
fig.canvas.mpl_connect('key_press_event', handle)
update()
input('')