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("(% 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)
print(' RectangleSelector activated.')
toggle_selector.RS.set_active(True)
# make a new plotting range
fig, current_ax = plt.subplots()
# If N is large one can see
N = 100000
# improvement by use blitting !
# plt.plot(x, +np.sin(.2 * np.pi * x),
# lw = 3.5, c ='b', alpha =.7)
# plot something
x = np.linspace(0.0, 10.0, N)
plt.plot(x, +np.cos(.2 * np.pi * x),
lw = 3.5, c ='c', alpha =.5)
plt.plot(x, -np.sin(.2 * np.pi * x),
lw = 3.5, c ='r', 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()