matplotlib部件之矩形选区(RectangleSelector)的实现

矩形选区概述

矩形选区是一种常见的对象选择方式，这个名词最常见于Photoshop中，用于在一个子图选择鼠标拖动的矩形区域中的元素，在matplotlib中的矩形选区属于部件（widgets），matplotlib中的部件都是中性（neutral ）的，即与具体后端实现无关。

矩形选区具体实现定义为matplotlib.widgets.RectangleSelector类，继承关系为：Widget->AxesWidget->_SelectorWidget->RectangleSelector。

RectangleSelector类的签名为class matplotlib.widgets.RectangleSelector(ax, onselect, drawtype='box', minspanx=0, minspany=0, useblit=False, lineprops=None, rectprops=None, spancoords='data', button=None, maxdist=10, marker_props=None, interactive=False, state_modifier_keys=None)

RectangleSelector类构造函数的参数为：

• ax：矩形选区生效的子图，类型为matplotlib.axes.Axes的实例。
• onselect：矩形选区完成后执行的回调函数，函数签名为def onselect(eclick: MouseEvent, erelease: MouseEvent)，eclick和erelease分别为开始和结束选区时的鼠标事件。
• drawtype：矩形选区的外观，取值范围为{"box", "line", "none"}，"box"为矩形框，"line"为矩形选区对角线，"none"无外观，类型为字符串，默认值为"box"。
• lineprops：当drawtype == "line"时线条的属性，默认值为dict(color="black", linestyle="-", linewidth=2, alpha=0.5)。
• rectprops：当drawtype == "box"时矩形框的属性，默认值为dict(facecolor="red", edgecolor="black", alpha=0.2, fill=True)。
• button：设置可用于触发矩形选区的鼠标键，MouseButton列表，默认为所有鼠标键。
• interactive：是否允许交互，布尔值，默认为False，即选择完成后选区即消失，值为True时，选区选择完成后不消失，除非按快捷键解除。
• state_modifier_keys：快捷键设置，类型为字典。
  • “move”： 移动已存在的选区，默认没有修饰键。
  • “clear”：清除现有选区，默认为 "escape"，即esc键。
  • “square”：正方形选区，默认为"shift"。
  • “center”：以当前点作为选区的中心点，默认为 "ctrl"。
  • “square” 和 “center” 可以组合使用。

案例

官方案例，https://matplotlib.org/gallery/widgets/rectangle_selector.html

案例说明

拖动鼠标画出矩形选区，默认为交互模式，显示选区框，按esc键取消选区，控制台显示选区的坐标和使用的鼠标键。按t键切换矩形选区功能的激活状态，非激活状态矩形选区功能不生效。

控制台输出：

(0.74, -0.38) --> (8.90, 0.75)
 The buttons you used were: 1 1

代码分析

from matplotlib.widgets import RectangleSelector
import numpy as np
import matplotlib.pyplot as plt

# 矩形选区选择时的回调函数
def line_select_callback(eclick, erelease):
  """
  Callback for line selection.

  *eclick* and *erelease* are the press and release events.
  """
  x1, y1 = eclick.xdata, eclick.ydata
  x2, y2 = erelease.xdata, erelease.ydata
  print(f"({x1:3.2f}, {y1:3.2f}) --> ({x2:3.2f}, {y2:3.2f})")
  print(f" The buttons you used were: {eclick.button} {erelease.button}")

# 激活状态快捷键回调函数，active属性和set_active方法继承自_SelectorWidget类
def toggle_selector(event):
  print(' Key pressed.')
  if event.key == 't':
    if RS.active:
      print(' RectangleSelector deactivated.')
      RS.set_active(False)
    else:
      print(' RectangleSelector activated.')
      RS.set_active(True)

# 绘图
fig, ax = plt.subplots()
N = 100000 # If N is large one can see improvement by using blitting.
x = np.linspace(0, 10, N)

ax.plot(x, np.sin(2*np.pi*x)) # plot something
ax.set_title(
  "Click and drag to draw a rectangle.\n"
  "Press 't' to toggle the selector on and off.")

# 构造矩形选区实例，选取外观为矩形框，鼠标键为左键右键有效，允许保留选区
# drawtype is 'box' or 'line' or 'none'
RS = RectangleSelector(ax, line_select_callback,
                    drawtype='box', useblit=True,
                    button=[1, 3], # disable middle button
                    minspanx=5, minspany=5,
                    spancoords='pixels',
                    interactive=True)
# 绑定键盘事件，实现切换矩形选区激活状态功能
fig.canvas.mpl_connect('key_press_event', toggle_selector)
plt.show()

到此这篇关于matplotlib部件之矩形选区(RectangleSelector)的实现的文章就介绍到这了,更多相关matplotlib 矩形选区内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！