Fix bug with non-numerical columns in util.aggregate

digitraceslab · Dec 5, 2023 · cb7b382 · cb7b382
1 parent 3751ba0
commit cb7b382
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/niimpy/preprocessing/util.py b/niimpy/preprocessing/util.py
@@ -192,11 +192,11 @@ def aggregate(df, freq, method_numerical='mean', method_categorical='first', gro
     assert method_numerical in ['mean', 'sum', 'median'], \
         'Cannot recognize sampling method. Possible values: "mean", "sum", "median".'
     if method_numerical == 'sum':
-        sub_df1 = groupby.resample(freq, **resample_kwargs).sum()
+        sub_df1 = groupby.resample(freq, **resample_kwargs).sum(numeric_only=True)
     elif method_numerical == 'mean':
-        sub_df1 = groupby.resample(freq, **resample_kwargs).mean()
+        sub_df1 = groupby.resample(freq, **resample_kwargs).mean(numeric_only=True)
     elif method_numerical == 'median':
-        sub_df1 = groupby.resample(freq, **resample_kwargs).median()
+        sub_df1 = groupby.resample(freq, **resample_kwargs).median(numeric_only=True)
     else:
         print("Can't recognize sampling method")