Merge pull request #2 from joulebit/overhaul

6% faster with better cropped cube

Author: bertie2

libraries/resizing.py

@@ -15,12 +15,11 @@ def crop_cube(cube: np.ndarray) -> np.ndarray:
     np.array: Cropped 3D Numpy byte array equivalent to cube, but with no zero padding
 
     """
-    for i in range(cube.ndim):
-        cube = np.swapaxes(cube, 0, i)
-        nonzero_indices = np.any(cube != 0, axis=tuple(range(1, cube.ndim)))
-        cube = cube[nonzero_indices]
-        cube = np.swapaxes(cube, 0, i)
-    return cube
+    nonzero_indices = np.where(cube != 0)
+    cropped_cube = cube[np.min(nonzero_indices[0]):np.max(nonzero_indices[0]) + 1,
+                         np.min(nonzero_indices[1]):np.max(nonzero_indices[1]) + 1,
+                         np.min(nonzero_indices[2]):np.max(nonzero_indices[2]) + 1]
+    return cropped_cube
 
 
 def expand_cube(cube: np.ndarray) -> Generator[np.ndarray, None, None]: