From 2c6dc05b9c3eb1f22def136340e1bc10a3b092ff Mon Sep 17 00:00:00 2001 From: richard Date: Sun, 16 Oct 2022 22:34:46 -0400 Subject: [PATCH] CLN: Replace _obj_with_exclusions with _select_obj in the case of a Series --- pandas/core/groupby/generic.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pandas/core/groupby/generic.py b/pandas/core/groupby/generic.py index 6296a79d13da9..7dbd0a609642e 100644 --- a/pandas/core/groupby/generic.py +++ b/pandas/core/groupby/generic.py @@ -181,7 +181,7 @@ def _wrap_agged_manager(self, mgr: Manager) -> Series: return ser def _get_data_to_aggregate(self) -> SingleManager: - ser = self._obj_with_exclusions + ser = self._selected_obj single = ser._mgr return single @@ -906,7 +906,7 @@ def nlargest( self, n: int = 5, keep: Literal["first", "last", "all"] = "first" ) -> Series: f = partial(Series.nlargest, n=n, keep=keep) - data = self._obj_with_exclusions + data = self._selected_obj # Don't change behavior if result index happens to be the same, i.e. # already ordered and n >= all group sizes. result = self._python_apply_general(f, data, not_indexed_same=True) @@ -917,7 +917,7 @@ def nsmallest( self, n: int = 5, keep: Literal["first", "last", "all"] = "first" ) -> Series: f = partial(Series.nsmallest, n=n, keep=keep) - data = self._obj_with_exclusions + data = self._selected_obj # Don't change behavior if result index happens to be the same, i.e. # already ordered and n >= all group sizes. result = self._python_apply_general(f, data, not_indexed_same=True)