edit

ml31415 · Jul 28, 2024 · 248a5ec · 248a5ec
1 parent fee245f
commit 248a5ec
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/numpy_groupies/aggregate_numpy.py b/numpy_groupies/aggregate_numpy.py
@@ -155,15 +155,21 @@ def _mean(group_idx, a, size, fill_value, dtype=np.dtype(np.float64)):
         ret = sums / counts
     if not np.isnan(fill_value):
         ret[counts == 0] = fill_value
-    return ret.astype(dtype, copy=False)
+    if iscomplexobj(a):
+        return ret
+    else:
+        ret.astype(dtype, copy=False)
 
 
 def _sum_of_squres(group_idx, a, size, fill_value, dtype=np.dtype(np.float64)):
     ret = np.bincount(group_idx, weights=a * a, minlength=size)
     if fill_value != 0:
         counts = np.bincount(group_idx, minlength=size)
         ret[counts == 0] = fill_value
-    return ret.astype(dtype, copy=False)
+    if iscomplexobj(a):
+        return ret
+    else:
+        ret.astype(dtype, copy=False)
 
 
 def _var(
@@ -183,7 +189,10 @@ def _var(
         ret = np.sqrt(ret)  # this is now std not var
     if not np.isnan(fill_value):
         ret[counts == 0] = fill_value
-    return ret.astype(dtype, copy=False)
+    if iscomplexobj(a):
+        return ret
+    else:
+        ret.astype(dtype, copy=False)
 
 
 def _std(group_idx, a, size, fill_value, dtype=np.dtype(np.float64), ddof=0):