文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

使用matplotlib怎么实现一个多边形选区功能

2023-06-06 13:15

关注

本篇文章为大家展示了使用matplotlib怎么实现一个多边形选区功能,内容简明扼要并且容易理解,绝对能使你眼前一亮,通过这篇文章的详细介绍希望你能有所收获。

多边形选区概述

多边形选区是一种常见的对象选择方式,在一个子图中,单击鼠标左键即构建一个多边形的端点,最后一个端点与第一个端点重合即完成多边形选区,选区即为多个端点构成的多边形。在matplotlib中的多边形选区属于部件(widgets),matplotlib中的部件都是中性(neutral )的,即与具体后端实现无关。

多边形选区具体实现定义为matplotlib.widgets.PolygonSelector类,继承关系为:Widget->AxesWidget->_SelectorWidget->PolygonSelector。

PolygonSelector类的签名为class matplotlib.widgets.PolygonSelector(ax, onselect, useblit=False, lineprops=None, markerprops=None, vertex_select_radius=15)

PolygonSelector类构造函数的参数为:

PolygonSelector类中的state_modifier_keys公有变量 state_modifier_keys定义了操作快捷键,类型为字典。

PolygonSelector类中的verts特性返回多边形选区中的多有端点,类型为列表,元素为(x,y)元组,即端点的坐标元组。

案例

官方案例,https://matplotlib.org/gallery/widgets/polygon_selector_demo.html

案例说明

使用matplotlib怎么实现一个多边形选区功能

单击鼠标左键创建端点,最终点击初始端点闭合多边形,形成多边形选区。选区外的数据元素颜色变淡,选区内数据颜色保持不变。

按esc键取消选区。按shift键鼠标可以移动多边形选区位置,按ctrl键鼠标可以移动多边形选区某个端点的位置。退出程序时,控制台输出选区内数据元素的坐标。

控制台输出:

Selected points:
[[2.0 2.0]
 [1.0 3.0]
 [2.0 3.0]]

案例代码

import numpy as npfrom matplotlib.widgets import PolygonSelectorfrom matplotlib.path import Pathclass SelectFromCollection:  """  Select indices from a matplotlib collection using `PolygonSelector`.  Selected indices are saved in the `ind` attribute. This tool fades out the  points that are not part of the selection (i.e., reduces their alpha  values). If your collection has alpha < 1, this tool will permanently  alter the alpha values.  Note that this tool selects collection objects based on their *origins*  (i.e., `offsets`).  Parameters  ----------  ax : `~matplotlib.axes.Axes`    Axes to interact with.  collection : `matplotlib.collections.Collection` subclass    Collection you want to select from.  alpha_other : 0 <= float <= 1    To highlight a selection, this tool sets all selected points to an    alpha value of 1 and non-selected points to *alpha_other*.  """  def __init__(self, ax, collection, alpha_other=0.3):    self.canvas = ax.figure.canvas    self.collection = collection    self.alpha_other = alpha_other    self.xys = collection.get_offsets()    self.Npts = len(self.xys)    # Ensure that we have separate colors for each object    self.fc = collection.get_facecolors()    if len(self.fc) == 0:      raise ValueError('Collection must have a facecolor')    elif len(self.fc) == 1:      self.fc = np.tile(self.fc, (self.Npts, 1))    self.poly = PolygonSelector(ax, self.onselect)    self.ind = []  def onselect(self, verts):    path = Path(verts)    self.ind = np.nonzero(path.contains_points(self.xys))[0]    self.fc[:, -1] = self.alpha_other    self.fc[self.ind, -1] = 1    self.collection.set_facecolors(self.fc)    self.canvas.draw_idle()  def disconnect(self):    self.poly.disconnect_events()    self.fc[:, -1] = 1    self.collection.set_facecolors(self.fc)    self.canvas.draw_idle()if __name__ == '__main__':  import matplotlib.pyplot as plt  fig, ax = plt.subplots()  grid_size = 5  grid_x = np.tile(np.arange(grid_size), grid_size)  grid_y = np.repeat(np.arange(grid_size), grid_size)  pts = ax.scatter(grid_x, grid_y)  selector = SelectFromCollection(ax, pts)  print("Select points in the figure by enclosing them within a polygon.")  print("Press the 'esc' key to start a new polygon.")  print("Try holding the 'shift' key to move all of the vertices.")  print("Try holding the 'ctrl' key to move a single vertex.")  plt.show()  selector.disconnect()  # After figure is closed print the coordinates of the selected points  print('\nSelected points:')  print(selector.xys[selector.ind])

上述内容就是使用matplotlib怎么实现一个多边形选区功能,你们学到知识或技能了吗?如果还想学到更多技能或者丰富自己的知识储备,欢迎关注编程网行业资讯频道。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯