minor cleanup in collections_demo.py

twmr · twmr · commit 64dfd4ae7700 · 2014-08-28T15:32:40.000+02:00
diff --git a/examples/api/collections_demo.py b/examples/api/collections_demo.py
@@ -26,8 +26,8 @@
 npts = 100
 
 # Make some spirals
-r = np.array(range(nverts))
-theta = np.array(range(nverts))*(2*np.pi)/(nverts - 1)
+r = np.arange(nverts)
+theta = np.linspace(0, 2*np.pi, nverts)
 xx = r * np.sin(theta)
 yy = r * np.cos(theta)
 spiral = list(zip(xx, yy))
