Merge pull request #5127 from ericmjl/mep12-cursor_demo.py

jenshnielsen · jenshnielsen · commit 96ed5c4a7be1 · 2015-09-29T09:42:38.000+01:00
mep12 on cursor_demo.py
diff --git a/examples/pylab_examples/cursor_demo.py b/examples/pylab_examples/cursor_demo.py
@@ -1,17 +1,19 @@
-#!/usr/bin/env python
 # -*- noplot -*-
 
 """
-
 This example shows how to use matplotlib to provide a data cursor.  It
 uses matplotlib to draw the cursor and may be a slow since this
 requires redrawing the figure with every mouse move.
 
 Faster cursoring is possible using native GUI drawing, as in
-wxcursor_demo.py
+wxcursor_demo.py.
+
+Also, mpldatacursor can be used to achieve a similar effect. See webpage:
+https://github.com/joferkington/mpldatacursor
 """
 from __future__ import print_function
-from pylab import *
+import matplotlib.pyplot as plt
+import numpy as np
 
 
 class Cursor(object):
@@ -33,7 +35,7 @@ def mouse_move(self, event):
         self.ly.set_xdata(x)
 
         self.txt.set_text('x=%1.2f, y=%1.2f' % (x, y))
-        draw()
+        plt.draw()
 
 
 class SnaptoCursor(object):
@@ -58,7 +60,7 @@ def mouse_move(self, event):
 
         x, y = event.xdata, event.ydata
 
-        indx = searchsorted(self.x, [x])[0]
+        indx = np.searchsorted(self.x, [x])[0]
         x = self.x[indx]
         y = self.y[indx]
         # update the line positions
@@ -67,16 +69,16 @@ def mouse_move(self, event):
 
         self.txt.set_text('x=%1.2f, y=%1.2f' % (x, y))
         print('x=%1.2f, y=%1.2f' % (x, y))
-        draw()
+        plt.draw()
 
-t = arange(0.0, 1.0, 0.01)
-s = sin(2*2*pi*t)
-fig, ax = subplots()
+t = np.arange(0.0, 1.0, 0.01)
+s = np.sin(2*2*np.pi*t)
+fig, ax = plt.subplots()
 
-cursor = Cursor(ax)
-#cursor = SnaptoCursor(ax, t, s)
-connect('motion_notify_event', cursor.mouse_move)
+#cursor = Cursor(ax)
+cursor = SnaptoCursor(ax, t, s)
+plt.connect('motion_notify_event', cursor.mouse_move)
 
 ax.plot(t, s, 'o')
-axis([0, 1, -1, 1])
-show()
+plt.axis([0, 1, -1, 1])
+plt.show()
