Блог

В R есть функция locator , похожая на Matlab ginput , где вы можете щелкнуть мышью по фигуре и выбрать любую координату x, y. Кроме того, существует вызываемая функция, в identify(x,y) которой, если вы зададите ей набор точек x, y, которые вы нанесли на график, а затем щелкните по рисунку, он вернет индекс точки x, y, которая находится ближе всего (в пределах регулируемого допуска) к выбранному вами местоположению (или несколькиминдексы, если выбрано несколько точек). Есть ли такая функциональность в Matplotlib?

Возможно, вы захотите использовать событие выбора :

 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)
 

Допуск, установленный параметром выбора там:

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

 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)
 

Вау, сколько лет прошло! Теперь matplotlib также поддерживается ginput функция, которая имеет почти тот же API, что и Matlab. Так что больше нет необходимости взламывать mpl-connect и так далее! (https://matplotlib.org/3.1.1/api/_as_gen/matplotlib.pyplot.ginput.html ) Например,,

 plt.ginput(4)
 

позволит пользователю выбрать 4 точки.

Это ginput() удобный инструмент для выбора координат x, y любой случайной точки из отображаемого окна, однако эта точка может не принадлежать отображаемым данным. Для выбора координат точки по x, y из построенных данных эффективным инструментом по-прежнему является использование 'pick_event' property with mpl_connect в качестве примера, приведенного в документации. Например:

 import matplotlib.pyplot as plt 
import numpy as np
from numpy.random import rand

fig, ax = plt.subplots()
ax.plot(rand(100), rand(100), picker=3)
# 3, for example, is tolerance for picker i.e, how far a mouse click from
# the plotted point can be registered to select nearby data point/points.

def on_pick(event):
    global points
    line = event.artist
    xdata, ydata = line.get_data()
    print('selected point is:',np.array([xdata[ind], ydata[ind]]).T)

cid = fig.canvas.mpl_connect('pick_event', on_pick)
 

Последняя строка выше соединит график с 'pick_event' и координаты ближайших точек графика будут продолжать печататься после каждого щелчка мыши на графике, чтобы завершить этот процесс, нам нужно использовать mpl_disconnect как:

 fig.canvas.mpl_disconnect(cid)