From 1908b791c17a771674826416192b4f9f8ac55d78 Mon Sep 17 00:00:00 2001 From: luenensc <peter.luenenschloss@ufz.de> Date: Sun, 3 Mar 2024 18:14:31 +0100 Subject: [PATCH] added pandas index support for setFlags --- saqc/funcs/flagtools.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/saqc/funcs/flagtools.py b/saqc/funcs/flagtools.py index 783ad6c2c..9e01e74e2 100644 --- a/saqc/funcs/flagtools.py +++ b/saqc/funcs/flagtools.py @@ -108,7 +108,7 @@ class FlagtoolsMixin: data : Determines which timestamps to set flags at, depending on the passed type: - * 1-d `array` or `List` of timestamps: flag `field` with `flag` at every timestamp in `f_data` + * 1-d `array` or `List` of timestamps or `pandas.Index`: flag `field` with `flag` at every timestamp in `f_data` * 2-d `array` or List of tuples: for all elements `t[k]` out of f_data: flag `field` with `flag` at every timestamp in between `t[k][0]` and `t[k][1]` * pd.Series: flag `field` with `flag` in between any index and data value of the passed series @@ -123,7 +123,7 @@ class FlagtoolsMixin: to_flag = pd.Series(False, index=self._data[field].index) # check if f_data is meant to denote timestamps: - if (isinstance(data, (list, np.ndarray))) and not isinstance( + if (isinstance(data, (list, np.ndarray, pd.Index))) and not isinstance( data[0], (tuple, np.ndarray) ): set_idx = pd.DatetimeIndex(data).intersection(to_flag.index) -- GitLab