如何实现plt.axhline生成的红色水平线上下拖动功能？

要让plt.axhline生成的红色水平线支持上下拖动，需要借助Matplotlib的事件处理机制，绑定鼠标的按下、移动和释放事件来实时更新线的位置。以下是完整的实现代码：

import matplotlib.pyplot as plt

# 替换成你自己的变量定义
frame_threshold = 5  # 原代码中的阈值
framenumber = range(100)
Avfiltered = [i*0.1 + 2 for i in framenumber]  # 示例数据，替换为你的真实数据

fig, ax = plt.subplots()
# 创建可选中的水平线，picker参数设置点击检测范围（像素）
drag_line = ax.axhline(y=frame_threshold, color='r', linestyle='-', picker=5)

# 标记拖动状态
is_dragging = False
active_line = None

def handle_mouse_press(event):
    global is_dragging, active_line
    # 确认点击目标是水平线且在当前坐标轴内
    if event.inaxes == ax and event.artist == drag_line:
        is_dragging = True
        active_line = event.artist

def handle_mouse_move(event):
    global is_dragging, active_line
    if is_dragging and event.inaxes == ax:
        # 更新水平线的y坐标
        active_line.set_ydata([event.ydata, event.ydata])
        fig.canvas.draw_idle()  # 异步刷新画布

def handle_mouse_release(event):
    global is_dragging, active_line
    is_dragging = False
    active_line = None

# 绑定鼠标事件到画布
fig.canvas.mpl_connect('button_press_event', handle_mouse_press)
fig.canvas.mpl_connect('motion_notify_event', handle_mouse_move)
fig.canvas.mpl_connect('button_release_event', handle_mouse_release)

# 绘制你的原始曲线
ax.plot(framenumber, Avfiltered)
plt.show()

关键说明： #

• picker=5：给水平线添加可选中属性，数值代表点击时的像素检测范围，确保能精准选中线条。
• 事件处理函数：分别监听鼠标按下、移动、释放三个动作，按下时标记开始拖动，移动时实时更新线的y坐标，释放时结束拖动状态。
• draw_idle()：相比直接draw()，它会等待合适时机刷新画布，避免频繁重绘导致卡顿。

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