代码之家 › 专栏 › 技术社区 › Mathieu Gauquelin

Matplotlib矩形选择器-设置初始位置

rectangles matplotlib python

Mathieu Gauquelin · 技术社区 · 7 年前

起初我认为我必须使用 state_modifier_keys 但似乎不是这样。

那么,有可能吗?

例如(有图但问题相同)可以是:

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


def line_select_callback(eclick, erelease):
    'eclick and erelease are the press and release events'
    x1, y1 = eclick.xdata, eclick.ydata
    x2, y2 = erelease.xdata, erelease.ydata
    print("(%3.2f, %3.2f) --> (%3.2f, %3.2f)" % (x1, y1, x2, y2))
    print(" The button you used were: %s %s" % (eclick.button, erelease.button))


def toggle_selector(event):
    print(' Key pressed.')
    if event.key in ['Q', 'q'] and toggle_selector.RS.active:
        print(' RectangleSelector deactivated.')
        toggle_selector.RS.set_active(False)
    if event.key in ['A', 'a'] and not toggle_selector.RS.active:
        print(' RectangleSelector activated.')
        toggle_selector.RS.set_active(True)


fig, current_ax = plt.subplots()                 # make a new plotting range
N = 100000                                       # If N is large one can see
x = np.linspace(0.0, 10.0, N)                    # improvement by use blitting!

plt.plot(x, +np.sin(.2*np.pi*x), lw=3.5, c='b', alpha=.7)  # plot something
plt.plot(x, +np.cos(.2*np.pi*x), lw=3.5, c='r', alpha=.5)
plt.plot(x, -np.sin(.2*np.pi*x), lw=3.5, c='g', alpha=.3)

print("\n      click  -->  release")

# drawtype is 'box' or 'line' or 'none'
toggle_selector.RS = RectangleSelector(current_ax, line_select_callback,
                                       drawtype='box', useblit=True,
                                       button=[1, 3],  # don't use middle button
                                       minspanx=5, minspany=5,
                                       spancoords='pixels',
                                       interactive=True)
plt.connect('key_press_event', toggle_selector)
plt.show()

2 回复 | 直到 7 年前

ImportanceOfBeingErnest 7 年前

没有blitting

如果你不愿意使用blitting,解决方案是设置 RectangleSelector 的补丁可见( RS.to_draw.set_visible(True) )并将其范围设置为您想要的数据( RS.extents = (0,4,0,1) ).

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


def line_select_callback(eclick, erelease):
    x1, y1 = eclick.xdata, eclick.ydata
    x2, y2 = erelease.xdata, erelease.ydata
    print("(%3.2f, %3.2f) --> (%3.2f, %3.2f)" % (x1, y1, x2, y2))

fig, current_ax = plt.subplots()
N = 100000 
x = np.linspace(0.0, 10.0, N) 

plt.plot(x, +np.sin(.2*np.pi*x), lw=3.5, c='b', alpha=.7)  # plot something
plt.plot(x, +np.cos(.2*np.pi*x), lw=3.5, c='r', alpha=.5)
plt.plot(x, -np.sin(.2*np.pi*x), lw=3.5, c='g', alpha=.3)


# drawtype is 'box' or 'line' or 'none'
RS = RectangleSelector(current_ax, line_select_callback,
                                       drawtype='box', useblit=False,
                                       button=[1, 3],  # don't use middle button
                                       minspanx=5, minspany=5,
                                       spancoords='pixels',
                                       interactive=True)

RS.to_draw.set_visible(True)
fig.canvas.draw()
RS.extents = (0,4,0,1)

plt.show()

带blitting

不幸的是,当使用blitting时,上述方法不起作用。
使用blitting的解决方案需要复制背景,然后将矩形设置为可见,但将 animated 属性设置为false。然后第一次绘制画布,显示矩形。但是,为了以后使用,我们需要设置 有生气的 属性再次设置为True。因此,首次在图形内部发生单击时,我们需要设置 有生气的 属性设置为True,并使用显示矩形之前已保存的背景对画布进行blit。

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


def line_select_callback(eclick, erelease):
    x1, y1 = eclick.xdata, eclick.ydata
    x2, y2 = erelease.xdata, erelease.ydata
    print("(%3.2f, %3.2f) --> (%3.2f, %3.2f)" % (x1, y1, x2, y2))

fig, current_ax = plt.subplots()
N = 100000 
x = np.linspace(0.0, 10.0, N) 

plt.plot(x, +np.sin(.2*np.pi*x), lw=3.5, c='b', alpha=.7)  # plot something
plt.plot(x, +np.cos(.2*np.pi*x), lw=3.5, c='r', alpha=.5)
plt.plot(x, -np.sin(.2*np.pi*x), lw=3.5, c='g', alpha=.3)


# drawtype is 'box' or 'line' or 'none'
RS = RectangleSelector(current_ax, line_select_callback,
                                       drawtype='box', useblit=True,
                                       button=[1, 3],  # don't use middle button
                                       minspanx=5, minspany=5,
                                       spancoords='pixels',
                                       interactive=True)
fig.canvas.draw()
bg = fig.canvas.copy_from_bbox(RS.ax.bbox)
RS.set_visible(True)

ext = (0,4,0,1)
RS.draw_shape(ext)
# Update displayed handles
RS._corner_handles.set_data(*RS.corners)
RS._edge_handles.set_data(*RS.edge_centers)
RS._center_handle.set_data(*RS.center)
for artist in RS.artists:
    RS.ax.draw_artist(artist)
    artist.set_animated(False)
fig.canvas.draw()

def initclick(evt):
    RS.background = bg
    RS.update()
    for artist in RS.artists:
        artist.set_animated(True)
    fig.canvas.mpl_disconnect(cid)

cid = fig.canvas.mpl_connect("button_press_event",initclick)

plt.show()

Oliver Zendel 3 年前

对于matplotlib v1.21.1和py3.7,这两个版本都有效(RS useblit=True和False):

RS.to_draw.set_visible(True)
RS.extents = (x1,x2,y1,y2)

注意矩形的坐标顺序:x1,x2,y1,y2(不是x1,y1,x2,y2!)