如何在matplotlib中使用Lasso部件

本篇文章給大家分享的是有關如何在matplotlib中使用Lasso部件，小編覺得挺實用的，因此分享給大家學習，希望大家閱讀完這篇文章后可以有所收獲，話不多說，跟著小編一起來看看吧。

套索概述

套索（Lasso）是與套索選區（LassoSelector）相似的matplotlib部件（widgets），兩者的區別主要在于：

繼承關系：

• 套索具體實現定義為matplotlib.widgets.Lasso類，繼承關系為：Widget->AxesWidget->Lasso。

• 套索選區具體實現定義為matplotlib.widgets.LassoSelector類，繼承關系為：Widget->AxesWidget->_SelectorWidget->LassoSelector。

構造參數：

• Lasso類的簽名為class matplotlib.widgets.Lasso(ax, xy, callback=None, useblit=True)，Lasso類需要給定套索一個起始的坐標。

• LassoSelector類的簽名為class matplotlib.widgets.LassoSelector(ax, onselect=None, useblit=True, lineprops=None, button=None)。
  

事件處理：

• Lasso事件在鼠標釋放時即被銷毀。

• LassoSelector在鼠標釋放時仍然可以繼續與子圖交互，直到斷開與子圖的連接。

Lasso類構造函數的參數為：

• ax：套索生效的子圖，類型為matplotlib.axes.Axes的實例。

• xy：套索起始的坐標。

• callback：套索完成即鼠標釋放時執行的回調函數，函數簽名為def func(verts)，verts的為套索端點的坐標列表。

套索可以使用matplotlib.path.Path類的contains_point方法獲取選區內的數據點。
貌似 Lasso是實驗性API，還不夠完善，matplotlib 3.3之后可能逐步廢棄 Lasso。

案例： 

官方案例，https://matplotlib.org/3.2.1/gallery/event_handling/lasso_demo.html

案例說明

案例代碼

from matplotlib import colors as mcolors, path
from matplotlib.collections import RegularPolyCollection
import matplotlib.pyplot as plt
from matplotlib.widgets import Lasso
import numpy as np


class Datum:
  colorin = mcolors.to_rgba("red")
  colorout = mcolors.to_rgba("blue")

  def __init__(self, x, y, include=False):
    self.x = x
    self.y = y
    if include:
      self.color = self.colorin
    else:
      self.color = self.colorout


class LassoManager:
  def __init__(self, ax, data):
    self.axes = ax
    self.canvas = ax.figure.canvas
    self.data = data

    self.Nxy = len(data)

    facecolors = [d.color for d in data]
    self.xys = [(d.x, d.y) for d in data]
    self.collection = RegularPolyCollection(
      6, sizes=(100,),
      facecolors=facecolors,
      offsets=self.xys,
      transOffset=ax.transData)

    ax.add_collection(self.collection)

    self.cid = self.canvas.mpl_connect('button_press_event', self.onpress)

  def callback(self, verts):
    facecolors = self.collection.get_facecolors()
    p = path.Path(verts)
    ind = p.contains_points(self.xys)
    for i in range(len(self.xys)):
      if ind[i]:
        facecolors[i] = Datum.colorin
      else:
        facecolors[i] = Datum.colorout

    self.canvas.draw_idle()
    self.canvas.widgetlock.release(self.lasso)
    del self.lasso

  def onpress(self, event):
    if self.canvas.widgetlock.locked():
      return
    if event.inaxes is None:
      return
    self.lasso = Lasso(event.inaxes,
              (event.xdata, event.ydata),
              self.callback)
    # acquire a lock on the widget drawing
    self.canvas.widgetlock(self.lasso)


if __name__ == '__main__':

  np.random.seed(19680801)

  data = [Datum(*xy) for xy in np.random.rand(100, 2)]
  ax = plt.axes(xlim=(0, 1), ylim=(0, 1), autoscale_on=False)
  ax.set_title('Lasso points using left mouse button')

  lman = LassoManager(ax, data)

  plt.show()

代碼分析

案例的關鍵代碼在于LassoManager類的onpress方法和callback方法。由于Lasso類在事件處理上比較原始，需要用戶進行控制，在鼠標按下、釋放事件中需要使用canvas.widgetlock對象鎖定/解鎖繪圖功能，保證只有一個對象進行繪圖，canvas.widgetlock是matplotlib.widgets.LockDraw類的實例。

以上就是如何在matplotlib中使用Lasso部件，小編相信有部分知識點可能是我們日常工作會見到或用到的。希望你能通過這篇文章學到更多知識。更多詳情敬請關注億速云行業資訊頻道。