matplotlib - tương tác chọn điểm hoặc địa điểm?

Question

Trong R, có chức năng locator giống như của Matlab ginput nơi bạn có thể nhấp vào hình bằng chuột và chọn bất kỳ tọa độ x, y nào. Ngoài ra, có một hàm gọi là identify(x,y), nếu bạn đặt cho nó một tập hợp các điểm x, y mà bạn đã vẽ và sau đó nhấp vào hình, nó sẽ trả về chỉ số của điểm x, y nằm gần nhất (trong một điều chỉnh dung sai) đến vị trí bạn đã chọn (hoặc nhiều chỉ mục, nếu chọn nhiều điểm). Có một chức năng như vậy trong Matplotlib?

Answer 1

Bạn có thể muốn sử dụng một pick event:

fig = figure() 
ax1 = fig.add_subplot(111) 
ax1.set_title('custom picker for line data') 
line, = ax1.plot(rand(100), rand(100), 'o', picker=line_picker) 
fig.canvas.mpl_connect('pick_event', onpick2) 

Tolerance được thiết lập bởi tham số bảng chọn có:

line, = ax1.plot(rand(100), 'o', picker=5) # 5 points tolerance 

Answer 2

from __future__ import print_function 
import matplotlib.pyplot as plt 
from matplotlib.lines import Line2D 
from matplotlib.patches import Rectangle 
from matplotlib.text import Text 
from matplotlib.image import AxesImage 
import numpy as np 
from numpy.random import rand 

if 1: 
    fig, ax = plt.subplots() 
    ax.set_title('click on points', picker=True) 
    ax.set_ylabel('ylabel', picker=True, bbox=dict(facecolor='red')) 
    line, = ax.plot(rand(100), 'o', picker=5) 

    def onpick1(event): 
     if isinstance(event.artist, Line2D): 
      thisline = event.artist 
      xdata = thisline.get_xdata() 
      ydata = thisline.get_ydata() 
      ind = event.ind 
      print 'X='+str(np.take(xdata, ind)[0]) # Print X point 
      print 'Y='+str(np.take(ydata, ind)[0]) # Print Y point 

    fig.canvas.mpl_connect('pick_event', onpick1)