BUG: Use linspace for coords in xarray_from_image

Prevent floating point precision issues from generating the incorrect number of coords.
InsightSoftwareConsortium · Apr 21, 2020 · 67ddcde · 67ddcde
1 parent 343cbf9
commit 67ddcde
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Wrapping/Generators/Python/itkExtras.py b/Wrapping/Generators/Python/itkExtras.py
@@ -405,10 +405,10 @@ def xarray_from_image(image):
     spatial_dims = ('x', 'y', 'z')
     coords = {}
     for index, dim in enumerate(spatial_dims[:spatial_dimension]):
-        coords[dim] = np.arange(origin[index],
-                                origin[index] + size[index]*spacing[index],
-                                spacing[index],
-                                dtype=np.float64)
+        coords[dim] = np.linspace(origin[index],
+                                  origin[index] + (size[index]-1)*spacing[index],
+                                  size[index],
+                                  dtype=np.float64)
 
     dims = list(reversed(spatial_dims[:spatial_dimension]))
     components = image.GetNumberOfComponentsPerPixel()