Merge pull request #10308 from anntzer/lasso-selector-no-input

tacaswell · web-flow · commit d7f69fe93c1e · 2018-01-23T18:42:24.000-05:00
DOC: Switch the lasso selector demo to use mpl event handling, not input().
diff --git a/examples/widgets/lasso_selector_demo_sgskip.py b/examples/widgets/lasso_selector_demo_sgskip.py
@@ -9,9 +9,8 @@
 a lasso loop around the points on the graph. To draw, just click
 on the graph, hold, and drag it around the points you need to select.
 """
-from __future__ import print_function
 
-from six.moves import input
+from __future__ import print_function
 
 import numpy as np
 
@@ -79,7 +78,6 @@ def disconnect(self):
 if __name__ == '__main__':
     import matplotlib.pyplot as plt
 
-    plt.ion()
     # Fixing random state for reproducibility
     np.random.seed(19680801)
 
@@ -91,11 +89,15 @@ def disconnect(self):
     pts = ax.scatter(data[:, 0], data[:, 1], s=80)
     selector = SelectFromCollection(ax, pts)
 
-    plt.draw()
-    input('Press Enter to accept selected points')
-    print("Selected points:")
-    print(selector.xys[selector.ind])
-    selector.disconnect()
+    def accept(event):
+        if event.key == "enter":
+            print("Selected points:")
+            print(selector.xys[selector.ind])
+            selector.disconnect()
+            ax.set_title("")
+            fig.canvas.draw()
+
+    fig.canvas.mpl_connect("key_press_event", accept)
+    ax.set_title("Press enter to accept selected points.")
 
-    # Block end of script so you can check that the lasso is disconnected.
-    input('Press Enter to quit')
+    plt.show()
