如何在Matplotlib中仅保留pick_event选中的最新标记点？

嘿，我来帮你搞定这个需求！你想要每次点击曲线时只显示最新的标记点，旧的自动消失对吧？先看看现有代码的小问题，再给你完整的解决方案：

核心解决思路 #

• 追踪当前标记点：用一个变量存储当前显示的标记点对象，初始设为None
• 清理旧标记：每次新点击时，先检查是否存在旧标记点，如果有就调用remove()把它从坐标轴上移除
• 只画一次原始曲线：原来的循环画10次曲线完全没必要，只会增加冗余和潜在问题
• 更新标记引用：绘制新标记后，把追踪变量更新为这个新的标记对象

修改后的可运行代码 #

import matplotlib.pyplot as plt
import numpy as np

fig, ax1 = plt.subplots()
t = np.arange(0.0, 1.0, 0.01)
s = np.sin(2 * 2 * np.pi * t)

# 只绘制一次原始曲线，开启picker支持点击
pt, = ax1.plot(t, s, picker=5)

# 用于保存当前标记点的变量，初始为空
current_marker = None

def onpick(event):
    global current_marker
    # 确保点击的是我们的目标曲线
    if event.artist != pt:
        return True
    if not len(event.ind):
        return True
    
    ind = event.ind[0]
    # 如果有旧标记，先移除它
    if current_marker is not None:
        current_marker.remove()
    # 绘制新的红色标记点，并保存这个对象
    current_marker, = ax1.plot(t[ind], s[ind], '|r', markersize=20)
    # 刷新画布，让修改立刻生效
    fig.canvas.draw()
    return True

fig.canvas.mpl_connect('pick_event', onpick)
plt.show()

代码细节说明 #

1. 移除冗余绘制：删掉了循环10次画曲线的代码，一次绘制就足够，避免重复的曲线对象干扰点击事件
2. 标记点追踪：current_marker变量全程记录当前显示的标记，每次新点击前都会清理旧标记
3. 正确获取标记对象：用, =接收plot的返回值（因为plot返回的是Line2D对象列表），这样才能调用remove()方法删除旧标记
4. 实时刷新画布：每次更新标记后调用fig.canvas.draw()，确保界面立刻显示最新状态

这样修改后，每次点击曲线只会保留最新的那个红色标记，旧标记会自动消失，完全符合你的需求！

内容的提问来源于stack exchange，提问作者r-tape