Skip to content
Snippets Groups Projects
Commit a56d8b73 authored by David Schäfer's avatar David Schäfer
Browse files

a new test function

parent ee9bd738
No related branches found
No related tags found
No related merge requests found
......@@ -13,6 +13,7 @@ def flagDispatch(func_name, *args, **kwargs):
"manflag": flagManual,
"mad": flagMad,
"constant": flagConstant,
"range": flagRange,
"generic": flagGeneric}
func = func_map.get(func_name, None)
......@@ -82,6 +83,14 @@ def flagManual(data, flags, field, flagger, **kwargs):
return data, flags
def flagRange(data, flags, field, flagger, min, max, **kwargs):
datacol = data[field].values
mask = (datacol < min) | (datacol >= max)
flags.loc[mask, field] = flagger.setFlag(flags.loc[mask, field], **kwargs)
return data, flags
def flagMad(data, flags, field, flagger, length, z, deriv, **kwargs):
def _flagMad(data: np.ndarray, z: int, deriv: int) -> np.ndarray:
......
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