Skip to content

Commit 6ee8fcb

Browse files
committed
fixup for int32
1 parent b79cc85 commit 6ee8fcb

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pandas/core/groupby/ops.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -786,6 +786,7 @@ def result_index_and_ids(self) -> tuple[Index, np.ndarray]:
786786
shape = tuple(len(level) for level in ob_levels)
787787
group_index = get_group_index(ob_codes, shape, sort=True, xnull=True)
788788
ob_ids, obs_group_ids = compress_group_index(group_index, sort=self._sort)
789+
ob_ids = ensure_platform_int(ob_ids)
789790
ob_index_codes = decons_obs_group_ids(
790791
ob_ids, obs_group_ids, shape, ob_codes, xnull=True
791792
)
@@ -833,9 +834,9 @@ def result_index_and_ids(self) -> tuple[Index, np.ndarray]:
833834
if self._sort:
834835
sorter = result_index.argsort()
835836
result_index = result_index.take(sorter)
836-
_, inverse = np.unique(sorter, return_index=True)
837+
_, index = np.unique(sorter, return_index=True)
837838
ids = ensure_platform_int(ids)
838-
ids = inverse.take(ids)
839+
ids = index.take(ids)
839840
else:
840841
ids, uniques = compress_group_index(ids, sort=False)
841842
ids = ensure_platform_int(ids)

0 commit comments

Comments
 (0)