Tidy up grid creation

ratt-ru · Apr 5, 2024 · 2d12ab7 · 2d12ab7
1 parent a8acad8
commit 2d12ab7
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/xarrayfits/grid.py b/xarrayfits/grid.py
@@ -29,11 +29,9 @@ def __init__(self, header: Mapping):
 
         self._grid = []
 
-        for d in range(ndims):
-            pixels = np.arange(1, self._naxis[d] + 1, dtype=np.float64)
-            self._grid.append(
-                (pixels - self._crpix[d]) * self._cdelt[d] + self._crval[d]
-            )
+        for na, rp, dt, rv in zip(self._naxis, self._crpix, self._cdelt, self._crval):
+            pixels = np.arange(1, na + 1, dtype=np.float64)
+            self._grid.append((pixels - rp) * dt + rv)
 
     @property
     def naxis(self):