Matplotlib子图拾取问题：无法将选中点数据分配到对应子图列表

需求与问题 #

需要实现一个Matplotlib拾取器，将不同子图中点击的点的x坐标按子图编号存入字典（键为子图索引，如0、1...），子图数量随数据集自动变化。当前能将标记字符串按子图存入嵌套列表，也能获取点数据存入扁平列表，但无法将点数据按子图分类存入嵌套结构。

失败的嵌套列表尝试 #

def test1(event):
    for i in list(range(len(fig.axes))):
        if event.inaxes == fig.axes[i]:
            test[i].append("click")

def test2(event):
    for i in list(range(len(fig.axes))):
        if event.inaxes == fig.axes[i]:
            thisline = event.artist
            xdata = thisline.get_xdata()
            ind = event.ind
            vals[i].append(xdata[ind])

x = np.random.rand(100)
y = np.random.rand(100)

fig, ax = plt.subplots(2,1)

test = [[] for i in fig.axes]
vals = [[] for i in fig.axes]

for ax in fig.axes:
    ax.plot(x, y, "o")
    fig.canvas.mpl_connect("button_press_event",test1)
    fig.canvas.mpl_connect("button_press_event",test2)

print(test)
print(vals)

点击后结果：

[['click', 'click', 'click', 'click'], ['click']] # 字符串存入正常
[[], []] # artist数据存入失败

成功但无法区分子图的扁平列表尝试 #

def onpick(event):
    thisline = event.artist
    xdata = thisline.get_xdata()
    ind = event.ind
    picker_tmp.append(xdata[ind])

fig, ax = plt.subplots(2,1)

picker_tmp = []

for ax in fig.axes:
    ax.plot(x, y, "o", picker=True)
    fig.canvas.mpl_connect('pick_event', onpick)

print(picker_tmp)

结果：

[array([0.249527]), array([0.36171572]), array([0.47993459]), array([0.8400918]), array([0.4248754])]

解决方案 #

问题核心：

1. 第一个示例使用button_press_event，该事件不包含artist和ind属性，无法获取点数据
2. 绘图时未添加picker=True，未启用元素拾取功能

正确实现代码：

import numpy as np
import matplotlib.pyplot as plt

# 初始化存储字典，自动适配子图数量
click_data = {}

def on_pick(event):
    # 获取点击所在的子图对象
    target_ax = event.inaxes
    # 获取子图在axes列表中的索引
    ax_index = fig.axes.index(target_ax)
    # 提取点击点的x坐标
    line = event.artist
    x_values = line.get_xdata()
    selected_indices = event.ind
    clicked_x = x_values[selected_indices][0]  # 单点击取第一个点，多选可直接存x_values[selected_indices]
    
    # 将数据存入字典，不存在的索引自动初始化列表
    if ax_index not in click_data:
        click_data[ax_index] = []
    click_data[ax_index].append(clicked_x)

# 生成测试数据
x = np.random.rand(100)
y = np.random.rand(100)

# 创建子图（数量可根据数据集调整）
fig, axes = plt.subplots(2, 1)

# 遍历子图，启用拾取并绑定事件
for ax in axes:
    ax.plot(x, y, "o", picker=True)  # 必须设置picker=True启用拾取
fig.canvas.mpl_connect('pick_event', on_pick)

plt.show()

# 输出按子图分类的结果
print("按子图存储的点击x坐标：")
for idx, coords in click_data.items():
    print(f"子图{idx}: {coords}")

关键说明 #

• 使用pick_event事件：该事件专门用于处理元素拾取，包含artist（被点击的绘图元素）和ind（点击点的索引）属性
• 启用拾取功能：绘图时添加picker=True，让散点可被识别为可拾取元素
• 子图索引匹配：通过event.inaxes获取点击的子图，再用fig.axes.index()得到其索引，实现按子图分类存储
• 字典存储：自动适配不同数量的子图，无需提前初始化固定长度的列表

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