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

[numpy] adds sigmaclip_mask

parent 2f64e6de
No related branches found
No related tags found
No related merge requests found
......@@ -87,3 +87,18 @@ def intersect2d(A, B, *args, **kwargs):
c = C
c = c.view(A.dtype).reshape(-1, ncols)
return C
def sigmaclip_mask(arr, low=4.0, high=4.0):
delta = 1
arr = np.array(arr)
mask = np.ones_like(arr, dtype=bool)
while delta:
arr_std = arr[mask].std()
arr_mean = arr[mask].mean()
good = np.sum(mask)
critlower = arr_mean - arr_std * low
critupper = arr_mean + arr_std * high
mask = mask & (arr >= critlower) & (arr <= critupper)
delta = good - np.sum(mask)
return ~mask
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