PREF: Fix regression from #58984 (#59025)

* check uses_mask before * backup gh * backup gh
pandas-dev · Jun 17, 2024 · 5d451fe · 5d451fe
1 parent b7970fc
commit 5d451fe
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pandas/_libs/groupby.pyx b/pandas/_libs/groupby.pyx
@@ -398,11 +398,12 @@ def group_cumsum(
         for i in range(N):
             lab = labels[i]
 
-            if lab < 0:
+            if uses_mask and lab < 0:
                 # GH#58811
-                if uses_mask:
-                    result_mask[i, :] = True
-                    out[i, :] = 0
+                result_mask[i, :] = True
+                out[i, :] = 0
+                continue
+            elif lab < 0:
                 continue
 
             for j in range(K):