Skip to content

Commit 81fcbc0

Browse files
committed
INT: use platform_int for np.repeat related (GH7975)
1 parent 5d4ce99 commit 81fcbc0

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

pandas/core/groupby.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2318,7 +2318,7 @@ def _transform_fast(self, func):
23182318
func = getattr(self,func)
23192319
values = func().values
23202320
counts = self.count().values
2321-
values = np.repeat(values, counts)
2321+
values = np.repeat(values, com._ensure_platform_int(counts))
23222322

23232323
# the values/counts are repeated according to the group index
23242324
indices = self.indices

pandas/tests/test_groupby.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -803,7 +803,8 @@ def test_transform_fast(self):
803803

804804
grp=df.groupby('id')['val']
805805

806-
expected = pd.Series(np.repeat(grp.mean().values, grp.count().values),index=df.index)
806+
values = np.repeat(grp.mean().values, com._ensure_platform_int(grp.count().values))
807+
expected = pd.Series(values,index=df.index)
807808
result = grp.transform(np.mean)
808809
assert_series_equal(result,expected)
809810

0 commit comments

Comments
 (0)