matplotlib對象拾取事件處理的實現

更新時間：2021年01月14日 08:35:58 作者：mighty13

這篇文章主要介紹了matplotlib對象拾取事件處理的實現，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

前面講到的眾多數據光標案例其實都用到了matplotlib的pick_event事件（拾取事件），該事件表示畫布中的對象被選中，該事件默認是禁用的，需要artist（數據元素，例如Line2D, Text, Patch, Polygon, AxesImage等）設置picker屬性才會啟用。

picker屬性

picker屬性有多種取值：

None：默認值，表示禁用 pick_event事件。
布爾值：如果為True，則啟用事件，當鼠標移動到該數據元素上方時觸發(fā)事件。
數值：如果取值是數值，則可以認為將該值賦給了pickradius（即事件的觸發(fā)半徑，默認值為15）。
函數：如果取值為函數，則可用于確定數據元素是否觸發(fā)鼠標事件。

pick_event事件

數據元素利用picker屬性啟用pick_event事件后，就可以使用事件管理器mpl_connect綁定事件。
pick_event事件除了常規(guī)屬性，還有兩個特有屬性：

mouseevent：生成拾取事件的鼠標事件。詳細信息請參閱matplotlib.backend_bases.MouseEvent。
artist：生成拾取事件的數據元素。

案例

下面以如下案例簡單說明pick_event事件的處理。
案例功能：觸發(fā)對象拾取事件則創(chuàng)建對象到兩個軸的垂線。
大致流程：

設置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()