to plt and np

Author: domspad

examples/pylab_examples/color_by_yvalue.py

@@ -1,18 +1,17 @@
 # use masked arrays to plot a line with different colors by y-value
-from numpy import logical_or, arange, sin, pi
-from numpy import ma
-from matplotlib.pyplot import plot, show
+import numpy as np
+import matplotlib.pyplot as plt
 
-t = arange(0.0, 2.0, 0.01)
-s = sin(2*pi*t)
+t = np.arange(0.0, 2.0, 0.01)
+s = np.sin(2*np.pi*t)
 
 upper = 0.77
 lower = -0.77
 
 
-supper = ma.masked_where(s < upper, s)
-slower = ma.masked_where(s > lower, s)
-smiddle = ma.masked_where(logical_or(s < lower, s > upper), s)
+supper = np.ma.masked_where(s < upper, s)
+slower = np.ma.masked_where(s > lower, s)
+smiddle = np.ma.masked_where(np.logical_or(s < lower, s > upper), s)
 
-plot(t, slower, 'r', t, smiddle, 'b', t, supper, 'g')
-show()
+plt.plot(t, slower, 'r', t, smiddle, 'b', t, supper, 'g')
+plt.show()