mep12 on plotfile_demo.py

ericmjl · ericmjl · commit 81928aafe12f · 2015-07-31T22:53:15.000-04:00
diff --git a/examples/pylab_examples/plotfile_demo.py b/examples/pylab_examples/plotfile_demo.py
@@ -1,35 +1,49 @@
-from pylab import plotfile, show, gca
+import matplotlib.pyplot as plt
+import numpy as np
+
 import matplotlib.cbook as cbook
 
 fname = cbook.get_sample_data('msft.csv', asfileobj=False)
 fname2 = cbook.get_sample_data('data_x_x2_x3.csv', asfileobj=False)
 
 # test 1; use ints
-plotfile(fname, (0, 5, 6))
+plt.plotfile(fname, (0, 5, 6))
 
 # test 2; use names
-plotfile(fname, ('date', 'volume', 'adj_close'))
+plt.plotfile(fname, ('date', 'volume', 'adj_close'))
 
 # test 3; use semilogy for volume
-plotfile(fname, ('date', 'volume', 'adj_close'), plotfuncs={'volume': 'semilogy'})
+plt.plotfile(
+    fname,
+    ('date',
+     'volume',
+     'adj_close'),
+    plotfuncs={
+        'volume': 'semilogy'})
 
 # test 4; use semilogy for volume
-plotfile(fname, (0, 5, 6), plotfuncs={5: 'semilogy'})
+plt.plotfile(fname, (0, 5, 6), plotfuncs={5: 'semilogy'})
 
 # test 5; single subplot
-plotfile(fname, ('date', 'open', 'high', 'low', 'close'), subplots=False)
+plt.plotfile(fname, ('date', 'open', 'high', 'low', 'close'), subplots=False)
 
 # test 6; labeling, if no names in csv-file
-plotfile(fname2, cols=(0, 1, 2), delimiter=' ',
-         names=['$x$', '$f(x)=x^2$', '$f(x)=x^3$'])
+plt.plotfile(fname2, cols=(0, 1, 2), delimiter=' ',
+             names=['$x$', '$f(x)=x^2$', '$f(x)=x^3$'])
 
 # test 7; more than one file per figure--illustrated here with a single file
-plotfile(fname2, cols=(0, 1), delimiter=' ')
-plotfile(fname2, cols=(0, 2), newfig=False, delimiter=' ')  # use current figure
-gca().set_xlabel(r'$x$')
-gca().set_ylabel(r'$f(x) = x^2, x^3$')
+plt.plotfile(fname2, cols=(0, 1), delimiter=' ')
+plt.plotfile(
+    fname2,
+    cols=(
+        0,
+        2),
+    newfig=False,
+    delimiter=' ')  # use current figure
+plt.xlabel(r'$x$')
+plt.ylabel(r'$f(x) = x^2, x^3$')
 
 # test 8; use bar for volume
-plotfile(fname, (0, 5, 6), plotfuncs={5: 'bar'})
+plt.plotfile(fname, (0, 5, 6), plotfuncs={5: 'bar'})
 
-show()
+plt.show()
