make it run

certik · May 29, 2009 · 7023268 · 7023268
1 parent 1df648a
commit 7023268
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/readbin3.py b/readbin3.py
@@ -1,3 +1,7 @@
+from math import pi
+
+from numpy import arange, meshgrid, sin, cos
+
 nplanets = 1
 nr = 800
 nphi = 4*nr
@@ -8,9 +12,12 @@
 dp = 2.*pi/nphi
 hdp = dp/2.0
 
-r, phi = meshgrid(0.4+dr*(1:nr)-hdr, 0+dp*(0:nphi)+hdp)
-x = r.*cos(phi)
-y = r.*sin(phi)
-radi = 0.4+dr*(1:nr)-hdr
+X = 0.4+dr*arange(1, nr+1)-hdr
+Y = 0+dp*arange(nphi+1)+hdp
+r, phi = meshgrid(X, Y)
+
+x = r * cos(phi)
+y = r * sin(phi)
+radi = X
 
 beta = 1.5  # power law for density profile