Small example fixes

dstansby · dstansby · commit f8ac7931ae6d · 2017-05-24T11:37:12.000+01:00
diff --git a/examples/color/color_demo.py b/examples/color/color_demo.py
@@ -22,9 +22,10 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-fig, ax = plt.subplots(facecolor='darkslategray')
 t = np.arange(0.0, 2.0, 0.01)
 s = np.sin(2 * np.pi * t)
+
+fig, ax = plt.subplots(facecolor='darkslategray')
 ax.plot(t, s, 'C1')
 ax.set_xlabel('time (s)', color='C1')
 ax.set_ylabel('voltage (mV)', color='0.5')  # grayscale color
diff --git a/examples/lines_bars_and_markers/scatter_custom_symbol.py b/examples/lines_bars_and_markers/scatter_custom_symbol.py
@@ -5,16 +5,15 @@
 
 """
 import matplotlib.pyplot as plt
-from numpy import arange, pi, cos, sin
-from numpy.random import rand
+import numpy as np
 
 # unit area ellipse
 rx, ry = 3., 1.
-area = rx * ry * pi
-theta = arange(0, 2 * pi + 0.01, 0.1)
-verts = list(zip(rx / area * cos(theta), ry / area * sin(theta)))
+area = rx * ry * np.pi
+theta = np.arange(0, 2 * np.pi + 0.01, 0.1)
+verts = list(zip(rx / area * np.cos(theta), ry / area * np.sin(theta)))
 
-x, y, s, c = rand(4, 30)
+x, y, s, c = np.random.rand(4, 30)
 s *= 10**2.
 
 fig, ax = plt.subplots()
diff --git a/examples/lines_bars_and_markers/scatter_star_poly.py b/examples/lines_bars_and_markers/scatter_star_poly.py
@@ -25,7 +25,7 @@
 plt.subplot(323)
 plt.scatter(x, y, s=80, c=z, marker=(verts, 0))
 # equivalent:
-# plt.scatter(x,y,s=80, c=z, marker=None, verts=verts)
+# plt.scatter(x, y, s=80, c=z, marker=None, verts=verts)
 
 plt.subplot(324)
 plt.scatter(x, y, s=80, c=z, marker=(5, 1))
