Skip to content

Commit 27cc0e2

Browse files
committed
FIX: Preserve DataFrame subclass type in groupby().agg()
1 parent 078e11f commit 27cc0e2

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

pandas/core/groupby/generic.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1585,6 +1585,9 @@ def aggregate(self, func=None, *args, engine=None, engine_kwargs=None, **kwargs)
15851585
result = self._insert_inaxis_grouper(result)
15861586
result.index = default_index(len(result))
15871587

1588+
if isinstance(self.obj, type(self.obj)):
1589+
result = self.obj._constructor(result)
1590+
15881591
return result
15891592

15901593
agg = aggregate

0 commit comments

Comments
 (0)