MEP12 chagnes on nan_test.py

ericmjl · ericmjl · commit 4a0259647c4f · 2015-07-28T14:25:44.000-04:00
diff --git a/examples/pylab_examples/nan_test.py b/examples/pylab_examples/nan_test.py
@@ -1,31 +1,29 @@
-#!/usr/bin/env python
 """
 Example: simple line plots with NaNs inserted.
 """
-from pylab import *
 import numpy as np
+import matplotlib.pyplot as plt
 
-t = arange(0.0, 1.0 + 0.01, 0.01)
-s = cos(2*2*pi*t)
-t[41:60] = NaN
-#t[50:60] = np.inf
+t = np.arange(0.0, 1.0 + 0.01, 0.01)
+s = np.cos(2*2*np.pi*t)
+t[41:60] = np.nan
 
-subplot(2, 1, 1)
-plot(t, s, '-', lw=2)
+plt.subplot(2, 1, 1)
+plt.plot(t, s, '-', lw=2)
 
-xlabel('time (s)')
-ylabel('voltage (mV)')
-title('A sine wave with a gap of NaNs between 0.4 and 0.6')
-grid(True)
+plt.xlabel('time (s)')
+plt.ylabel('voltage (mV)')
+plt.title('A sine wave with a gap of NaNs between 0.4 and 0.6')
+plt.grid(True)
 
-subplot(2, 1, 2)
-t[0] = NaN
-t[-1] = NaN
-plot(t, s, '-', lw=2)
-title('Also with NaN in first and last point')
+plt.subplot(2, 1, 2)
+t[0] = np.nan
+t[-1] = np.nan
+plt.plot(t, s, '-', lw=2)
+plt.title('Also with NaN in first and last point')
 
-xlabel('time (s)')
-ylabel('more nans')
-grid(True)
+plt.xlabel('time (s)')
+plt.ylabel('more nans')
+plt.grid(True)
 
-show()
+plt.show()
