如何通过按键终止Matplotlib实时绘图的while循环？

方法一：用Matplotlib内置按键事件（无需额外库） #

直接基于你现有的matplotlib代码修改，通过绑定键盘事件触发循环终止，按下q键就会关闭窗口并停止循环，完全不影响实时绘图的流畅性。

修改后的示例代码：

import matplotlib.pyplot as plt

plt.ion()
x_data = []
y_data = []
graph, = plt.plot(x_data, y_data)  # 注意加逗号，plot返回的是列表，要取出单个对象

# 标记是否终止循环
stop_flag = False

def on_key_press(event):
    global stop_flag
    # 按下q键触发终止
    if event.key == 'q':
        stop_flag = True
        plt.close()

# 绑定按键事件到绘图窗口
plt.connect('key_press_event', on_key_press)

while not stop_flag:
    # 这里替换成你从网络分析仪获取数据的逻辑
    # 示例：模拟数据更新
    x_data.append(len(x_data))
    y_data.append(len(y_data)**0.5)
    
    graph.set_ydata(y_data)
    graph.set_xdata(x_data)
    plt.xlim(0, len(x_data))  # 自动调整x轴范围，避免数据超出显示
    plt.ylim(0, max(y_data)+1 if y_data else 1)  # 自动调整y轴范围
    plt.draw()
    plt.pause(0.01)

print("绘图循环已终止")

关键点说明 #

• 用plt.connect绑定按键回调函数，只要绘图窗口处于焦点，按下q就会触发终止
• graph, = plt.plot(...)必须加逗号，否则后续调用set_xdata/set_ydata会报错
• 全局变量stop_flag控制循环的运行状态，无需等待输入，不影响实时绘图效率

方法二：用pynput监听全局键盘（支持窗口失焦触发） #

如果需要在绘图窗口不是当前焦点时也能按键终止，可以用pynput库，先在树莓派上安装：

pip install pynput

示例代码：

import matplotlib.pyplot as plt
from pynput import keyboard

plt.ion()
x_data = []
y_data = []
graph, = plt.plot(x_data, y_data)
stop_flag = False

def on_press(key):
    global stop_flag
    try:
        # 按下q键终止
        if key.char == 'q':
            stop_flag = True
            plt.close()
            return False  # 停止键盘监听线程
    except AttributeError:
        # 处理ESC等特殊按键
        if key == keyboard.Key.esc:
            stop_flag = True
            plt.close()
            return False

# 启动独立的键盘监听线程
listener = keyboard.Listener(on_press=on_press)
listener.start()

while not stop_flag:
    # 替换为你的数据获取逻辑
    x_data.append(len(x_data))
    y_data.append(len(y_data)**0.5)
    
    graph.set_ydata(y_data)
    graph.set_xdata(x_data)
    plt.xlim(0, len(x_data))
    plt.ylim(0, max(y_data)+1 if y_data else 1)
    plt.draw()
    plt.pause(0.01)

listener.join()
print("绘图循环已终止")

关键点说明 #

• pynput会启动独立线程监听键盘，完全不阻塞主绘图循环
• 支持监听普通字符键和特殊按键（如ESC），返回False可停止监听
• 适合需要在后台也能触发终止的场景

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