本篇文章给大家分享的是有关如何在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, pathfrom matplotlib.collections import RegularPolyCollectionimport matplotlib.pyplot as pltfrom matplotlib.widgets import Lassoimport numpy as npclass 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.coloroutclass 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部件，小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注编程网行业资讯频道。
