matplotlibでマウスのイベントを取得して、クリックした点のデータをとってくる。
参考URLを勉強がてら少しシンプルにしてみる。
描画データをPointBrowserに関連付けておいて、データをクリックするたびに、その上に黄色い薄い丸を描画するようにしている。
メモ：
plotのpicker=5でクリックしたときの許容範囲を決定。
複数のデータがクリック範囲に入ったときも、そのうち１番最初の１つのみを使う。(ind[0])
mpl_connectを使えば、マウス以外にもイベントを取得できる。ハンドラーというのだろう。(http://matplotlib.sourceforge.net/api/backend_bases_api.html#matplotlib.backend_bases.FigureCanvasBase.mpl_connect)
参考URL：http://matplotlib.sourceforge.net/examples/event_handling/data_browser.html
from numpy.random import rand
from pylab import figure, show
class PointBrowser:
"""
pass a subplot to __init__
overwrite the yellow circle on the clicked data
"""
def __init__(self, fig, ax):
self.fig = fig
self.text = ax.text(0.05, 0.95, 'selected: none',
transform=ax.transAxes, va='top')
self.selected, = ax.plot([], [], 'o', ms=12, alpha=0.4,
color='yellow', visible=False)
def onpick(self, event):
"""
it picks up the index of clicked data
"""
if not len(event.ind):
return True
ind = event.ind[0]
x = event.artist.get_xdata()[ind]
y = event.artist.get_ydata()[ind]
clicked = (x, y)
self.selected.set_visible(True)
self.selected.set_data(clicked)
self.text.set_text(' x: %f\n y: %f'%(clicked))
self.fig.canvas.draw()
if __name__ == "__main__":
xs = rand(100)
ys = rand(100)
fig = figure()
ax = fig.add_subplot(111)
ax.set_title('picking demo')
ax.plot(xs, ys, 'o', picker=5) # 5 points tolerance
browser = PointBrowser(fig, ax)
fig.canvas.mpl_connect('pick_event', browser.onpick)
show()
