DOC: fixing/cleanup of units scatter example

phobson · phobson · commit 92a23cc5e8bd · 2016-08-18T17:37:47.000-07:00
diff --git a/examples/units/units_scatter.py b/examples/units/units_scatter.py
@@ -13,14 +13,13 @@
 from matplotlib import pyplot
 
 # create masked array
+data = (1, 2, 3, 4, 5, 6, 7, 8)
+mask = (1, 0, 1, 0, 0, 0, 1, 0)
+xsecs = secs * np.ma.MaskedArray(data, mask, float)
 
-
-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, (ax1, ax2, ax3) = pyplot.figure(nrows=3, sharex=True)
-ax1 = fig.add_subplot(3, 1, 1)
+fig, (ax1, ax2, ax3) = pyplot.subplots(nrows=3, sharex=True)
 ax1.scatter(xsecs, xsecs)
+ax1.yaxis.set_units(secs)
 ax1.axis([0, 10, 0, 10])
 
 ax2.scatter(xsecs, xsecs, yunits=hertz)
@@ -30,4 +29,5 @@
 ax3.yaxis.set_units(minutes)
 ax3.axis([0, 10, 0, 1])
 
-fig.show()
+fig.tight_layout()
+pyplot.show()
