ENH: move answers into sub-directory

tacaswell · tacaswell · commit 6edcb4756b67 · 2017-11-06T00:16:31.000-05:00
diff --git a/key/00-explore.py b/key/00-explore.py
@@ -0,0 +1,30 @@
+import matplotlib.pyplot as plt
+import numpy as np
+plt.ion()
+
+last_ev = None
+
+
+def event_printer(event):
+    """Helper function for exploring events.
+
+    Prints all public attributes +
+    """
+    # capture the last event
+    global last_ev
+    last_ev = event
+    if event.inaxes is not None:
+        print(f'{event.name} ({event.xdata:.3f}, {event.ydata:.3f})')
+    else:
+        print(f'{event.name}')
+
+
+th = np.linspace(0, 2*np.pi, 64)
+fig, ax = plt.subplots()
+ax.plot(th, np.sin(th), 'o-', picker=5)
+
+cids = {k: fig.canvas.mpl_connect(k, event_printer)
+        for k in ('button_press_event', 'button_release_event',
+                  'scroll_event',
+                  'key_press_event', 'key_release_event',
+                  'pick_event')}
diff --git a/key/01-callable.py b/key/01-callable.py
@@ -39,7 +39,7 @@ def collect_string(self):
         return ''.join([ev.key for ev in self.event_deque
                         if ev.name == 'key_press_event'])
 
+
 fc = FormatterCollector()
-cid = fig.canvas.mpl_connect('key_press_event', fc)
-cid2 = fig.canvas.mpl_connect('key_release_event', fc)
-cid2 = fig.canvas.mpl_connect('button_press_event', fc)
+cids = {k: fig.canvas.mpl_connect(k, fc)
+        for k in ('key_press_event')}
diff --git a/key/02-event_filter.py b/key/02-event_filter.py
@@ -5,7 +5,7 @@
 
 
 class LineMaker:
-    def __init__(self, ln):
+    def __init__(self, ln, *, color_cycle=None):
         # stash the current data
         self.xdata = list(ln.get_xdata())
         self.ydata = list(ln.get_ydata())
@@ -73,6 +73,15 @@ def on_key(self, event):
             # ask the GUI to re-draw the next time it can
             self.ln.figure.canvas.draw_idle()
 
+        try:
+            lw = int(event.key)
+        except (ValueError):
+            pass
+        else:
+            self.ln.set_linewidth(lw)
+            self.ln.figure.canvas.draw_idle()
+
+
 fig, ax = plt.subplots()
 ln, = ax.plot([], [], '-o')
 line_maker = LineMaker(ln)
