Make examples that load npy files with datetime(objects) py3 compatible

jenshnielsen · jenshnielsen · commit ca40ff5a4cbd · 2015-10-21T12:22:08.000+01:00
Requires np 1.10 to work under py3
diff --git a/examples/api/date_demo.py b/examples/api/date_demo.py
@@ -26,7 +26,14 @@
 # The record array stores python datetime.date as an object array in
 # the date column
 datafile = cbook.get_sample_data('goog.npy')
-r = np.load(datafile).view(np.recarray)
+try:
+    # Python3 cannot load python2 .npy files with datetime(object) arrays
+    # unless the encoding is set to bytes. Hovever this option was
+    # not added until numpy 1.10 so this example will only work with
+    # python 2 or with numpy 1.10 and later.
+    r = np.load(datafile, encoding='bytes').view(np.recarray)
+except TypeError:
+    r = np.load(datafile).view(np.recarray)
 
 fig, ax = plt.subplots()
 ax.plot(r.date, r.adj_close)
diff --git a/examples/pylab_examples/centered_ticklabels.py b/examples/pylab_examples/centered_ticklabels.py
@@ -20,7 +20,14 @@
 
 # load some financial data; apple's stock price
 fh = cbook.get_sample_data('aapl.npy.gz')
-r = np.load(fh)
+try:
+    # Python3 cannot load python2 .npy files with datetime(object) arrays
+    # unless the encoding is set to bytes. Hovever this option was
+    # not added until numpy 1.10 so this example will only work with
+    # python 2 or with numpy 1.10 and later.
+    r = np.load(fh, encoding='bytes')
+except TypeError:
+    r = np.load(fh)
 fh.close()
 r = r[-250:]  # get the last 250 days
 
diff --git a/examples/pylab_examples/scatter_demo2.py b/examples/pylab_examples/scatter_demo2.py
@@ -10,7 +10,14 @@
 # The record array stores python datetime.date as an object array in
 # the date column
 datafile = cbook.get_sample_data('goog.npy')
-price_data = np.load(datafile).view(np.recarray)
+try:
+    # Python3 cannot load python2 .npy files with datetime(object) arrays
+    # unless the encoding is set to bytes. Hovever this option was
+    # not added until numpy 1.10 so this example will only work with
+    # python 2 or with numpy 1.10 and later
+    price_data = np.load(datafile, encoding='bytes').view(np.recarray)
+except TypeError:
+    price_data = np.load(datafile).view(np.recarray)
 price_data = price_data[-250:]  # get the most recent 250 trading days
 
 delta1 = np.diff(price_data.adj_close)/price_data.adj_close[:-1]
