fixup for int32

pandas-dev · Nov 14, 2023 · 6ee8fcb · 6ee8fcb
1 parent b79cc85
commit 6ee8fcb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pandas/core/groupby/ops.py b/pandas/core/groupby/ops.py
@@ -786,6 +786,7 @@ def result_index_and_ids(self) -> tuple[Index, np.ndarray]:
             shape = tuple(len(level) for level in ob_levels)
             group_index = get_group_index(ob_codes, shape, sort=True, xnull=True)
             ob_ids, obs_group_ids = compress_group_index(group_index, sort=self._sort)
+            ob_ids = ensure_platform_int(ob_ids)
             ob_index_codes = decons_obs_group_ids(
                 ob_ids, obs_group_ids, shape, ob_codes, xnull=True
             )
@@ -833,9 +834,9 @@ def result_index_and_ids(self) -> tuple[Index, np.ndarray]:
             if self._sort:
                 sorter = result_index.argsort()
                 result_index = result_index.take(sorter)
-                _, inverse = np.unique(sorter, return_index=True)
+                _, index = np.unique(sorter, return_index=True)
                 ids = ensure_platform_int(ids)
-                ids = inverse.take(ids)
+                ids = index.take(ids)
             else:
                 ids, uniques = compress_group_index(ids, sort=False)
                 ids = ensure_platform_int(ids)