Skip to content
Snippets Groups Projects
Commit 515a8354 authored by Dennis Noll's avatar Dennis Noll
Browse files

[tools] numpy: adds pad_until_along_axis

parent ccef9fa0
No related branches found
No related tags found
No related merge requests found
......@@ -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)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment