matplotlib对象拾取事件处理

前面讲到的众多数据光标案例其实都用到了matplotlib的pick_event事件（拾取事件），该事件表示画布中的对象被选中，该事件默认是禁用的，需要artist（数据元素，例如Line2D, Text, Patch, Polygon, AxesImage等）设置picker属性才会启用。

picker属性

picker属性有多种取值：

• None：默认值，表示禁用 pick_event事件。
• 布尔值：如果为True，则启用事件，当鼠标移动到该数据元素上方时触发事件。
• 数值：如果取值是数值，则可以认为将该值赋给了pickradius（即事件的触发半径，默认值为15）。
• 函数：如果取值为函数，则可用于确定数据元素是否触发鼠标事件。

pick_event事件

数据元素利用picker属性启用pick_event事件后，就可以使用事件管理器mpl_connect绑定事件。
pick_event事件除了常规属性，还有两个特有属性：

• mouseevent：生成拾取事件的鼠标事件。详细信息请参阅matplotlib.backend_bases.MouseEvent。
• artist：生成拾取事件的数据元素。

案例

下面以如下案例简单说明pick_event事件的处理。
案例功能：触发对象拾取事件则创建对象到两个轴的垂线。
大致流程：

• 设置picker属性，启用pick_event事件。
• 编写回调函数，详见注释。
• 绑定事件。

import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_subplot(111)
ax.set_title('click on points')
# 启用对象拾取事件
line, = ax.plot(np.random.rand(10)*10, 'o', picker=5)
ax.axis([0,10,0,10])

def onpick(event):
    # 通过event.artist属性获取被拾取到的数据元素
    thisline = event.artist
    # 获取数据元素的x,y坐标数组
    xdata = thisline.get_xdata()
    ydata = thisline.get_ydata()
    # 获取到当前元素的索引
    ind = event.ind
    # xdata[ind]为当前元素的x坐标，ydata[ind]为当前元素的y坐标。
    ax.plot((0,xdata[ind]), (ydata[ind],ydata[ind]), 'g--')
    ax.plot((xdata[ind],xdata[ind]), (0,ydata[ind]), 'g--')
    # 重绘图像
    thisline.figure.canvas.draw()
# 绑定事件
fig.canvas.mpl_connect('pick_event', onpick)

plt.show()