DOC: remove pylab from units example

phobson · phobson · commit 74f16b6056ba · 2016-08-18T16:33:51.000-07:00
diff --git a/examples/units/units_scatter.py b/examples/units/units_scatter.py
@@ -10,27 +10,24 @@
 """
 import numpy as np
 from basic_units import secs, hertz, minutes
-from matplotlib.pylab import figure, show
+from matplotlib import pyplot
 
 # create masked array
 
 
 xsecs = secs*np.ma.MaskedArray((1, 2, 3, 4, 5, 6, 7, 8), (1, 0, 1, 0, 0, 0, 1, 0), float)
 #xsecs = secs*np.arange(1,10.)
 
-fig = figure()
+fig, (ax1, ax2, ax3) = pyplot.figure(nrows=3, sharex=True)
 ax1 = fig.add_subplot(3, 1, 1)
 ax1.scatter(xsecs, xsecs)
-#ax1.set_ylabel('seconds')
 ax1.axis([0, 10, 0, 10])
 
-ax2 = fig.add_subplot(3, 1, 2, sharex=ax1)
 ax2.scatter(xsecs, xsecs, yunits=hertz)
 ax2.axis([0, 10, 0, 1])
 
-ax3 = fig.add_subplot(3, 1, 3, sharex=ax1)
 ax3.scatter(xsecs, xsecs, yunits=hertz)
 ax3.yaxis.set_units(minutes)
 ax3.axis([0, 10, 0, 1])
 
-show()
+fig.show()
