matplotlib.lines.VertexSelector

class matplotlib.lines.VertexSelector(line)

Управляет обратными вызовами для поддержания списка выбранных вершин для matplotlib.lines.Line2D. Производные классы должны переопределить process_selected(), чтобы выполнить какие-либо действия с выделенными элементами.

Вот пример, который выделяет выбранные вершины красными кружками:

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.lines as lines

class HighlightSelected(lines.VertexSelector):
    def __init__(self, line, fmt='ro', **kwargs):
        lines.VertexSelector.__init__(self, line)
        self.markers, = self.axes.plot([], [], fmt, **kwargs)

    def process_selected(self, ind, xs, ys):
        self.markers.set_data(xs, ys)
        self.canvas.draw()

fig = plt.figure()
ax = fig.add_subplot(111)
x, y = np.random.rand(2, 30)
line, = ax.plot(x, y, 'bs-', picker=5)

selector = HighlightSelected(line)
plt.show()

Инициализируйте класс с экземпляром matplotlib.lines.Line2D. Линия должна быть уже добавлена к некоторому экземпляру matplotlib.axes.Axes и должна иметь свойство пикера, установленное.

onpick(event)

При выделении линии обновляется набор выделенных индексов.

process_selected(ind, xs, ys)

Реализация по умолчанию «ничего не делать» метода process_selected().

ind — это индексы выбранных вершин. xs и ys — координаты выбранных вершин.