aboutsummaryrefslogtreecommitdiffstats
path: root/astro
diff options
context:
space:
mode:
authorAaron LI <aly@aaronly.me>2017-07-11 20:37:56 +0800
committerAaron LI <aly@aaronly.me>2017-07-11 20:37:56 +0800
commita391a79a54b8661e690a1d74a88d6599a8ff3f53 (patch)
treeb6a9423f00cdef6b8904a32115b143bce13a9cf4 /astro
parentb22deb9783de362997975ff7f778a443cba48d34 (diff)
downloadatoolbox-a391a79a54b8661e690a1d74a88d6599a8ff3f53.tar.bz2
astro/ps2d.py: Fix a bug/typo
Signed-off-by: Aaron LI <aly@aaronly.me>
Diffstat (limited to 'astro')
-rwxr-xr-xastro/ps2d.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/astro/ps2d.py b/astro/ps2d.py
index b6b5cc8..e110776 100755
--- a/astro/ps2d.py
+++ b/astro/ps2d.py
@@ -132,7 +132,7 @@ class PS2D:
ic_x = (np.abs(k_x) < eps).nonzero()[0][0]
ic_z = (np.abs(k_z) < eps).nonzero()[0][0]
p_x = np.arange(nx) - ic_x
- p_z = np.abs(np.arange(ny) - ic_z)
+ p_z = np.abs(np.arange(nz) - ic_z)
mx, my = np.meshgrid(p_x, p_x)
rho, phi = self.cart2pol(mx, my)
rho = np.around(rho).astype(np.int)