Skip to content

Commit a76b3f4

Browse files
authored
CLN: Replace _selected_obj with _obj_with_exclusions in SeriesGroupBy (#55392)
1 parent 940a1b4 commit a76b3f4

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

pandas/core/groupby/generic.py

+7-7
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ def _wrap_agged_manager(self, mgr: Manager) -> Series:
158158
def _get_data_to_aggregate(
159159
self, *, numeric_only: bool = False, name: str | None = None
160160
) -> SingleManager:
161-
ser = self._selected_obj
161+
ser = self._obj_with_exclusions
162162
single = ser._mgr
163163
if numeric_only and not is_numeric_dtype(ser.dtype):
164164
# GH#41291 match Series behavior
@@ -448,7 +448,7 @@ def _aggregate_named(self, func, *args, **kwargs):
448448
initialized = False
449449

450450
for name, group in self.grouper.get_iterator(
451-
self._selected_obj, axis=self.axis
451+
self._obj_with_exclusions, axis=self.axis
452452
):
453453
# needed for pandas/tests/groupby/test_groupby.py::test_basic_aggregations
454454
object.__setattr__(group, "name", name)
@@ -519,7 +519,7 @@ def _cython_transform(
519519
):
520520
assert axis == 0 # handled by caller
521521

522-
obj = self._selected_obj
522+
obj = self._obj_with_exclusions
523523

524524
try:
525525
result = self.grouper._cython_operation(
@@ -546,7 +546,7 @@ def _transform_general(
546546

547547
results = []
548548
for name, group in self.grouper.get_iterator(
549-
self._selected_obj, axis=self.axis
549+
self._obj_with_exclusions, axis=self.axis
550550
):
551551
# this setattr is needed for test_transform_lambda_with_datetimetz
552552
object.__setattr__(group, "name", name)
@@ -618,7 +618,7 @@ def true_and_notna(x) -> bool:
618618
indices = [
619619
self._get_index(name)
620620
for name, group in self.grouper.get_iterator(
621-
self._selected_obj, axis=self.axis
621+
self._obj_with_exclusions, axis=self.axis
622622
)
623623
if true_and_notna(group)
624624
]
@@ -1164,7 +1164,7 @@ def nlargest(
11641164
self, n: int = 5, keep: Literal["first", "last", "all"] = "first"
11651165
) -> Series:
11661166
f = partial(Series.nlargest, n=n, keep=keep)
1167-
data = self._selected_obj
1167+
data = self._obj_with_exclusions
11681168
# Don't change behavior if result index happens to be the same, i.e.
11691169
# already ordered and n >= all group sizes.
11701170
result = self._python_apply_general(f, data, not_indexed_same=True)
@@ -1175,7 +1175,7 @@ def nsmallest(
11751175
self, n: int = 5, keep: Literal["first", "last", "all"] = "first"
11761176
) -> Series:
11771177
f = partial(Series.nsmallest, n=n, keep=keep)
1178-
data = self._selected_obj
1178+
data = self._obj_with_exclusions
11791179
# Don't change behavior if result index happens to be the same, i.e.
11801180
# already ordered and n >= all group sizes.
11811181
result = self._python_apply_general(f, data, not_indexed_same=True)

0 commit comments

Comments
 (0)