快捷導(dǎo)航

matplotlib對象拾取事件處理的實現(xiàn)

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

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

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

picker屬性

picker屬性有多種取值：

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

pick_event事件

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

mouseevent：生成拾取事件的鼠標(biāo)事件。詳細(xì)信息請參閱matplotlib.backend_bases.MouseEvent。
artist：生成拾取事件的數(shù)據(jù)元素。

案例

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

設(shè)置picker屬性，啟用pick_event事件。
編寫回調(diào)函數(shù)，詳見注釋。
綁定事件。

import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_subplot(111)
ax.set_title('click on points')
# 設(shè)置繪圖元素的picker屬性啟用對象拾取事件
line, = ax.plot(np.random.rand(10)*10, 'o', picker=5)
ax.axis([0,10,0,10])

def onpick(event):
  # 通過event.artist屬性獲取被拾取到的數(shù)據(jù)元素
  thisline = event.artist
  # 獲取數(shù)據(jù)元素的x,y坐標(biāo)數(shù)組
  xdata = thisline.get_xdata()
  ydata = thisline.get_ydata()
  # 獲取到當(dāng)前元素的索引
  ind = event.ind
  # xdata[ind]為當(dāng)前元素的x坐標(biāo)，ydata[ind]為當(dāng)前元素的y坐標(biāo)。
  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()