matplotlib对象拾取事件处理的实现

(编辑：jimmy 日期: 2024/11/2 浏览：3 次 )

前面讲到的众多数据光标案例其实都用到了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')
# 设置绘图元素的picker属性启用对象拾取事件
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()

上一篇：详解如何使用Pytest进行自动化测试
下一篇：用python查找统一局域网下ip对应的mac地址