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):

print(" The button you used were: %s %s" % (eclick.button, erelease.button))

def toggle_selector(event):

toggle_selector.RS.set_active(True)

fig, current_ax = plt.subplots() # make a new plotingrange

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")

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()