matplotlib部件（widgets）之椭圆形选区（EllipseSelector）

椭圆形选区概述

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

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

matplotlib.widgets.EllipseSelector类为RectangleSelector类的子类，仅重写了和形状相关的若干方法，因此两个类实例很多特性都非常相似。

EllipseSelector类的签名为class matplotlib.widgets.EllipseSelector(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)

EllipseSelector类构造函数的参数为：

• 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” 可以组合使用。

案例

案例说明

案例创建了两个子图，一个子图drawtype="box"，一个子图drawtype="line"。拖动鼠标画出椭圆形选区，设置交互模式，显示选区框，按esc键取消选区。。

源代码

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

def onselect(eclick, erelease):
    "eclick and erelease are matplotlib events at press and release."
    pass

x = np.arange(100.) / 99
y = np.sin(x)
fig, (ax1,ax2) = plt.subplots(2)
ax1.plot(x, y)
ax2.plot(x, y)

selector1 = EllipseSelector(ax1, onselect, drawtype='box',interactive=True)
selector2 = EllipseSelector(ax2, onselect, drawtype='line',interactive=True)
plt.show()