File tree Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -2318,7 +2318,7 @@ def _transform_fast(self, func):
2318
2318
func = getattr (self ,func )
2319
2319
values = func ().values
2320
2320
counts = self .count ().values
2321
- values = np .repeat (values , counts )
2321
+ values = np .repeat (values , com . _ensure_platform_int ( counts ) )
2322
2322
2323
2323
# the values/counts are repeated according to the group index
2324
2324
indices = self .indices
Original file line number Diff line number Diff line change @@ -803,7 +803,8 @@ def test_transform_fast(self):
803
803
804
804
grp = df .groupby ('id' )['val' ]
805
805
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 )
807
808
result = grp .transform (np .mean )
808
809
assert_series_equal (result ,expected )
809
810
You can’t perform that action at this time.
0 commit comments