Skip to content

Commit a477dc0

Browse files
committed
Call ensure_plantform_int last
1 parent 99d2beb commit a477dc0

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

pandas/core/groupby/ops.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -762,7 +762,7 @@ def ids(self) -> np.ndarray:
762762
@cache_readonly
763763
def result_index_and_ids(self) -> tuple[Index, np.ndarray]:
764764
names = self.names
765-
codes = [ensure_platform_int(ping.codes) for ping in self.groupings]
765+
codes = [ping.codes for ping in self.groupings]
766766
levels = [Index._with_infer(ping.uniques) for ping in self.groupings]
767767
obs = [
768768
ping._observed or not ping._passed_categorical for ping in self.groupings
@@ -807,10 +807,10 @@ def result_index_and_ids(self) -> tuple[Index, np.ndarray]:
807807

808808
if all(obs):
809809
result_index = ob_index
810-
ids = ob_ids
810+
ids = ensure_platform_int(ob_ids)
811811
elif not any(obs):
812812
result_index = unob_index
813-
ids = unob_ids
813+
ids = ensure_platform_int(unob_ids)
814814
else:
815815
# Combine unobserved and observed parts of result_index
816816
unob_indices = [k for k, e in enumerate(obs) if not e]
@@ -841,6 +841,7 @@ def result_index_and_ids(self) -> tuple[Index, np.ndarray]:
841841
[uniques, np.delete(np.arange(len(result_index)), uniques)]
842842
)
843843
result_index = result_index.take(taker)
844+
ids = ensure_platform_int(ids)
844845

845846
return result_index, ids
846847

0 commit comments

Comments
 (0)