如何在Matplotlib中模拟触发pick事件的鼠标点击操作

我们需要实现simulate_pick(ax, x, y)函数，当给定的(x,y)数据坐标足够接近线条顶点时，自动触发对应的pick事件处理逻辑（移除线条），让Matplotlib自动识别符合条件的artist，无需手动指定。

实现思路 #

1. 将输入的(x,y)数据坐标转换为Matplotlib的显示坐标（像素坐标系），因为picker的距离判断基于此。
2. 遍历当前axes中的所有线条对象。
3. 对每条线条，提取其顶点的坐标并转换为显示坐标，计算每个顶点与模拟点击位置的欧氏距离。
4. 如果存在顶点与点击位置的距离小于线条设置的picker阈值，就创建对应的PickEvent对象，调用remove函数处理该线条。

完整代码示例 #

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.backend_bases import PickEvent, MouseEvent

def remove(event):
    event.artist.remove()
    # 移除后刷新画布
    event.canvas.draw()

def simulate_pick(ax, x, y):
    # 将数据坐标转换为显示坐标（像素）
    click_pos = ax.transData.transform((x, y))
    
    # 遍历axes中的所有线条artist
    for artist in ax.get_children():
        if not isinstance(artist, plt.Line2D):
            continue
        # 获取线条的顶点数据坐标
        line_x, line_y = artist.get_data()
        # 转换为显示坐标
        line_positions = ax.transData.transform(np.column_stack((line_x, line_y)))
        # 计算每个顶点与点击位置的距离
        distances = np.linalg.norm(line_positions - click_pos, axis=1)
        # 获取线条的picker阈值，默认是None，这里用5.0作为 fallback
        picker_threshold = artist.get_picker() or 5.0
        
        # 如果有顶点距离小于阈值，触发pick事件
        if np.any(distances <= picker_threshold):
            # 模拟鼠标事件（仅用于创建PickEvent）
            mock_mouse_event = MouseEvent(
                'button_press_event', ax.figure.canvas, 
                click_pos[0], click_pos[1], 1, 0,  # 模拟左键点击
                xdata=x, ydata=y
            )
            # 创建PickEvent，ind是距离最近的顶点索引
            ind = np.argmin(distances)
            pick_event = PickEvent(
                'pick_event', ax.figure.canvas, mock_mouse_event, 
                artist, ind=ind
            )
            # 调用remove函数处理
            remove(pick_event)
            # 找到符合条件的线条后退出，避免一次移除多条（可根据需求调整）
            break

# 测试代码
N = 10
fig, ax = plt.subplots()
for i in range(N):
    line, = ax.plot(np.random.random(2), np.random.random(2), '-o', picker=5.0)

fig.canvas.mpl_connect('pick_event', remove)

# 模拟点击某个线条的顶点位置（这里取第一条线的第一个顶点）
first_line = ax.get_lines()[0]
sim_x, sim_y = first_line.get_data()[0][0], first_line.get_data()[1][0]
simulate_pick(ax, sim_x, sim_y)

plt.show()

关键细节说明 #

• 坐标转换：ax.transData.transform()负责将数据坐标（图表上的数值）转换为显示坐标（画布的像素位置），确保距离计算的一致性。
• Picker阈值：通过artist.get_picker()获取线条设置的picker参数，若未设置则使用默认的5.0像素阈值。
• 事件模拟：创建MouseEvent和PickEvent对象，完全模拟Matplotlib原生pick事件的触发逻辑，让remove函数可以直接复用。

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