Merge pull request #4846 from ericmjl/mep12_plotfile_demo.py

jenshnielsen · jenshnielsen · commit 54bdd3e817d1 · 2015-08-06T09:30:53.000+01:00
mep12 on plotfile_demo.py
diff --git a/examples/pylab_examples/plotfile_demo.py b/examples/pylab_examples/plotfile_demo.py
@@ -1,35 +1,39 @@
-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()
