From 578245fea92c15421cf2f3a64706519a0e0fd2b0 Mon Sep 17 00:00:00 2001 From: luenensc <peter.luenenschloss@ufz.de> Date: Sat, 9 Mar 2024 11:32:05 +0100 Subject: [PATCH] additions --- saqc/funcs/generic.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/saqc/funcs/generic.py b/saqc/funcs/generic.py index 56c00bb25..7f85016a0 100644 --- a/saqc/funcs/generic.py +++ b/saqc/funcs/generic.py @@ -251,7 +251,10 @@ class GenericMixin: dfilter = kwargs.get("dfilter", BAD) dchunk, fchunk = self._data[fields].copy(), self._flags[fields].copy() + trg_idx = dchunk[dchunk.columns[0]].index result = _execGeneric(fchunk, dchunk, func, dfilter=dfilter) + result = _inferBroadcast(result, (len(trg_idx), len(targets))) + result = _inferDF(result, cols=targets, index=trg_idx) result = _castResult(result) if len(result.columns) > 1 and len(targets) != len(result.columns): -- GitLab