调用matplotlib.pyplot waitforbuttonpress()时如何获取按键值？

当然有办法区分不同的按键操作啦！plt.waitforbuttonpress()本身只返回布尔值，但我们可以结合matplotlib的事件系统，轻松获取到用户按下的具体按键，同时完全保留缩放、平移这些鼠标交互功能。

下面给你两种实用的实现方式：

方法一：结合waitforbuttonpress()和last_key_event #

当waitforbuttonpress()返回True时，说明触发了键盘事件，这时候我们可以通过当前图形画布的last_key_event属性拿到完整的键盘事件对象，从中提取按下的按键信息。

示例代码：

import matplotlib.pyplot as plt

# 绘制示例图形
plt.plot([1, 3, 2, 4])
plt.show(block=False)  # 非阻塞显示，保证鼠标交互正常

while True:
    # 等待按键，鼠标操作（缩放、平移）会被正常处理
    if plt.waitforbuttonpress():
        # 获取最近的键盘事件
        key_event = plt.gcf().canvas.last_key_event
        pressed_key = key_event.key
        print(f"检测到按下按键: {pressed_key}")
        
        # 根据不同按键执行对应操作
        if pressed_key == 'q':
            print("触发退出操作，程序结束")
            break
        elif pressed_key == 's':
            print("触发保存操作，图形已保存为plot.png")
            plt.savefig('plot.png')
        elif pressed_key == 'r':
            print("触发重置视图操作")
            plt.gca().relim()
            plt.gca().autoscale_view()
            plt.draw()

方法二：自定义键盘事件回调函数 #

如果需要更灵活的持续监听按键（不需要循环阻塞），可以直接给图形画布绑定键盘事件回调函数，这样既能响应按键，又不影响任何鼠标交互操作。

示例代码：

import matplotlib.pyplot as plt

def handle_key_press(event):
    pressed_key = event.key
    print(f"检测到按下按键: {pressed_key}")
    
    # 定义不同按键的操作逻辑
    if pressed_key == 'q':
        plt.close()
        print("程序已退出")
    elif pressed_key == 's':
        plt.savefig('plot.png')
        print("图形已成功保存")
    elif pressed_key == '+' or pressed_key == '=':
        # 手动放大Y轴范围
        current_ylim = plt.gca().get_ylim()
        plt.gca().set_ylim(current_ylim[0], current_ylim[1] * 0.9)
        plt.draw()
    elif pressed_key == '-':
        # 手动缩小Y轴范围
        current_ylim = plt.gca().get_ylim()
        plt.gca().set_ylim(current_ylim[0], current_ylim[1] * 1.1)
        plt.draw()

# 绘制示例图形
plt.plot([1, 3, 2, 4])
# 绑定键盘事件回调
plt.gcf().canvas.mpl_connect('key_press_event', handle_key_press)

# 阻塞显示，保持交互
plt.show()

小提示 #

• 特殊按键的表示：回车键是'enter'，空格键是' '，方向键分别是'up'、'down'、'left'、'right'，你可以通过打印pressed_key来查看所有按键的对应标识。
• 两种方法都完全兼容matplotlib的默认鼠标交互（缩放、平移等），不会造成冲突。

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