PyPlot：使用picker高亮选中的数据点

我来帮你搞定这个选中点高亮的需求！你的核心目标是每次只让当前选中的点变成黄色，之前选中的自动恢复蓝色对吧？咱们先看原代码的小局限：你用plot画的蓝色圆点是一个整体的Line2D对象，所有点共享同一个颜色属性，没法单独修改某个点的颜色。咱们换成scatter来绘制点，就能实现单个点的颜色控制了。

下面是修改后的完整代码，我会在后面拆解关键改动：

import matplotlib.pyplot as plt

class MyPlot(object):
    def __init__(self, parent=None):
        super(self.__class__, self).__init__()
        self.selected_ind = None  # 记录上一次选中的点索引

    def makePlot(self):
        self.fig = plt.figure('Test', figsize=(10, 8))
        ax = plt.subplot(111)
        x = range(0, 100, 10)
        y = (5,)*10
        ax.plot(x, y, '-', color='red')
        # 用scatter画点，每个点初始为蓝色，存储这个scatter对象以便后续修改
        self.points = ax.scatter(x, y, color='blue', picker=5)
        plt.connect('pick_event', self.onPick)
        plt.show()

    def onPick(self, event=None):
        # 确保只处理我们的点对象，避免响应其他元素的选中事件
        if event.artist != self.points:
            return
        
        ind = event.ind[0]  # 获取选中的第一个点的索引（支持多选时取第一个）
        colors = self.points.get_facecolors()  # 获取所有点的颜色数组

        # 恢复之前选中的点为蓝色
        if self.selected_ind is not None:
            colors[self.selected_ind] = [0, 0, 1, 1]  # RGBA格式的蓝色

        # 设置当前选中的点为黄色
        colors[ind] = [1, 1, 0, 1]  # RGBA格式的黄色
        self.points.set_facecolors(colors)  # 更新颜色数组
        self.selected_ind = ind  # 更新选中索引
        self.fig.canvas.draw()  # 刷新画布，让颜色变化生效

        # 保留你原有打印坐标的功能
        x_value = self.points.get_offsets()[ind][0]
        y_value = self.points.get_offsets()[ind][1]
        print(f'x:{x_value}')
        print(f'y:{y_value}')

if __name__ == '__main__':
    app = MyPlot()
    app.makePlot()

关键改动说明 #

1. 用scatter替代plot画点：
   scatter会为每个点生成独立的颜色控制属性，而plot的点是一个整体对象，无法单独修改单个点的颜色。

2. 维护选中状态：
   添加self.selected_ind属性记录上一次选中的点索引，每次选中新点时先恢复之前点的颜色，再高亮当前点。

3. RGBA颜色设置：
   matplotlib使用RGBA数组（四个0-1之间的数值）表示颜色，蓝色是[0,0,1,1]，黄色是[1,1,0,1]，最后一个值是透明度。

4. 刷新画布：
   修改颜色后必须调用self.fig.canvas.draw()才能让界面实时更新颜色变化。

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