Skip to content

Commit 9c89702

Browse files
authored
REF: Use _python_apply_general internally (#52342)
1 parent e80dc87 commit 9c89702

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

pandas/core/groupby/generic.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -2685,7 +2685,9 @@ def dtypes(self) -> Series:
26852685
)
26862686

26872687
# error: Incompatible return value type (got "DataFrame", expected "Series")
2688-
return self.apply(lambda df: df.dtypes) # type: ignore[return-value]
2688+
return self._python_apply_general( # type: ignore[return-value]
2689+
lambda df: df.dtypes, self._selected_obj
2690+
)
26892691

26902692
@doc(DataFrame.corrwith.__doc__)
26912693
def corrwith(

pandas/core/groupby/groupby.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -673,14 +673,14 @@ def f(self):
673673
return self.plot(*args, **kwargs)
674674

675675
f.__name__ = "plot"
676-
return self._groupby.apply(f)
676+
return self._groupby._python_apply_general(f, self._groupby._selected_obj)
677677

678678
def __getattr__(self, name: str):
679679
def attr(*args, **kwargs):
680680
def f(self):
681681
return getattr(self.plot, name)(*args, **kwargs)
682682

683-
return self._groupby.apply(f)
683+
return self._groupby._python_apply_general(f, self._groupby._selected_obj)
684684

685685
return attr
686686

@@ -1117,7 +1117,7 @@ def curried(x):
11171117
# special case otherwise extra plots are created when catching the
11181118
# exception below
11191119
if name in base.plotting_methods:
1120-
return self.apply(curried)
1120+
return self._python_apply_general(curried, self._selected_obj)
11211121

11221122
is_transform = name in base.transformation_kernels
11231123
result = self._python_apply_general(

0 commit comments

Comments
 (0)