Tidy up grid creation

ratt-ru · Apr 5, 2024 · aa061ed · aa061ed
1 parent a8acad8
commit aa061ed
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/xarrayfits/grid.py b/xarrayfits/grid.py
@@ -29,11 +29,11 @@ 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 enumerate(
+            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):