added in get_two_stock_demo into file so no more imports

Salil Vanvari · Salil Vanvari · commit e8c4ef12fec3 · 2016-11-28T22:31:24.000-08:00
diff --git a/examples/pylab_examples/stock_demo.py b/examples/pylab_examples/stock_demo.py
@@ -2,7 +2,28 @@
 import numpy as np
 
 from matplotlib.ticker import MultipleLocator
-from data_helper import get_two_stock_data
+
+def get_two_stock_data():
+    """
+    load stock time and price data for two stocks The return values
+    (d1,p1,d2,p2) are the trade time (in days) and prices for stocks 1
+    and 2 (intc and aapl)
+    """
+    ticker1, ticker2 = 'INTC', 'AAPL'
+
+    file1 = cbook.get_sample_data('INTC.dat.gz')
+    file2 = cbook.get_sample_data('AAPL.dat.gz')
+    M1 = fromstring(file1.read(), '<d')
+
+    M1 = resize(M1, (M1.shape[0]//2, 2))
+
+    M2 = fromstring(file2.read(), '<d')
+    M2 = resize(M2, (M2.shape[0]//2, 2))
+
+    d1, p1 = M1[:, 0], M1[:, 1]
+    d2, p2 = M2[:, 0], M2[:, 1]
+    return (d1, p1, d2, p2)
+
 
 d1, p1, d2, p2 = get_two_stock_data()
 
@@ -11,7 +32,7 @@
 lines2 = plt.plot(d2, p2, 'r', label="AAPL")
 plt.xlabel('Days')
 plt.ylabel('Normalized price')
-plt.xlim(0, 3)
+plt.xlim(0, 3 )
 ax.xaxis.set_major_locator(MultipleLocator(1))
 
 plt.title('INTC vs AAPL')
