Skip to content

Commit 25770be

Browse files
committed
fixup
1 parent 3a9892d commit 25770be

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

pandas/core/groupby/ops.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@
2424
)
2525
import pandas._libs.groupby as libgroupby
2626
from pandas._typing import (
27-
AnyArrayLike,
2827
ArrayLike,
2928
AxisInt,
3029
NDFrameT,
@@ -767,7 +766,7 @@ def result_index_and_ids(self) -> tuple[Index, np.ndarray]:
767766

768767
codes_and_uniques = [ping._codes_and_uniques for ping in self.groupings]
769768

770-
codes = [e[0] for e in codes_and_uniques]
769+
codes = [e[0].astype("intp", copy=False) for e in codes_and_uniques]
771770
levels = [Index._with_infer(e[1]) for e in codes_and_uniques]
772771
for k, (ping, level) in enumerate(zip(self.groupings, levels)):
773772
if ping._passed_categorical:
@@ -780,9 +779,9 @@ def result_index_and_ids(self) -> tuple[Index, np.ndarray]:
780779
]
781780

782781
if len(self.groupings) == 1:
783-
result_index: AnyArrayLike = levels[0]
782+
result_index = levels[0]
784783
result_index.name = names[0]
785-
ids = codes[0].astype("intp")
784+
ids = codes[0]
786785
return result_index, ids
787786
elif any(obs):
788787
ob_codes = [e for e, o in zip(codes, obs) if o]

0 commit comments

Comments
 (0)