[tools] numpy: adds pad_until_along_axis

numpy.py

@@ -102,3 +102,22 @@ def sigmaclip_mask(arr, low=4.0, high=4.0):
         mask = mask & (arr >= critlower) & (arr <= critupper)
         delta = good - np.sum(mask)
     return ~mask
+
+
+def pad_until_along_axis(array: np.ndarray, target_length: int, axis: int = 0):
+    """Add zeros along one axis until target_length is reached
+
+    Args:
+        array (np.ndarray): Array to pad
+        target_length (int): Lenght until which to pad
+        axis (int, optional): Axis along which to pad
+
+    Returns:
+        np.ndarray: Padded array
+    """
+    pad_size = target_length - array.shape[axis]
+    if pad_size <= 0:
+        return array
+    npad = [(0, 0)] * array.ndim
+    npad[axis] = (0, pad_size)
+    return np.pad(array, pad_width=npad, mode="constant", constant_values=0)