在Python Matplotlib中通过按键事件切换水平边界线显示

要实现按特定按键切换水平线的显示和隐藏，核心是保存水平线的对象引用，通过这个引用来控制它的可见性或者直接移除。以下是两种可行的实现方案：

方案一：切换水平线的可见性（推荐） #

通过保存axhline()返回的线条对象，每次按键时切换其visible属性，实现显示/隐藏的切换：

import matplotlib.pyplot as plt

# 创建画布和轴对象，绘制示例数据
fig, ax = plt.subplots()
ax.plot([1,2,3,4], [1,4,2,3])

boundary = 2.5  # 你的边界值
boundary_line = None  # 用于保存水平线的引用

def on_key(event):
    global boundary_line
    if event.key == 'b':
        if boundary_line is None:
            # 首次按b键，创建水平线并保存引用
            boundary_line = ax.axhline(y=boundary, color='green', linestyle='-')
        else:
            # 再次按b键，切换水平线的可见状态
            boundary_line.set_visible(not boundary_line.get_visible())
        # 重绘画布使更改生效
        fig.canvas.draw()

# 绑定键盘事件回调
fig.canvas.mpl_connect('key_press_event', on_key)

plt.show()

你之前尝试布尔变量控制失败，大概率是因为没有正确保存线条对象的引用，或者修改可见性后没有重绘画布。

方案二：直接移除/重建水平线 #

如果需要彻底移除水平线而非隐藏，可以在再次按键时调用remove()方法删除线条，重置引用后下次按键重新创建：

import matplotlib.pyplot as plt

fig, ax = plt.subplots()
ax.plot([1,2,3,4], [1,4,2,3])

boundary = 2.5
boundary_line = None

def on_key(event):
    global boundary_line
    if event.key == 'b':
        if boundary_line is None:
            boundary_line = ax.axhline(y=boundary, color='green', linestyle='-')
        else:
            # 移除水平线并重置引用
            boundary_line.remove()
            boundary_line = None
        fig.canvas.draw()

fig.canvas.mpl_connect('key_press_event', on_key)
plt.show()

单独按键实现移除功能 #

如果需要用另一个按键（比如r键）单独移除水平线，可以在回调函数中添加分支：

def on_key(event):
    global boundary_line
    if event.key == 'b':
        if boundary_line is None:
            boundary_line = ax.axhline(y=boundary, color='green', linestyle='-')
        else:
            boundary_line.set_visible(not boundary_line.get_visible())
        fig.canvas.draw()
    elif event.key == 'r':
        # 按r键移除水平线（如果存在）
        if boundary_line is not None:
            boundary_line.remove()
            boundary_line = None
            fig.canvas.draw()

关键注意点 #

• 一定要保存axhline()返回的对象，这是后续操作的唯一依据
• 任何更改后都需要调用fig.canvas.draw()重绘画布，否则界面不会更新
• 切换可见性比反复删除重建更高效，适合频繁切换的场景

内容的提问来源于stack exchange，提问作者blair.smyth