Skip to content

Commit 9bb2b87

Browse files
committed
CLN: Apply.agg_list_like
1 parent afb0bf8 commit 9bb2b87

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

pandas/core/apply.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -323,6 +323,7 @@ def agg_or_apply_list_like(
323323

324324
is_groupby = isinstance(obj, (DataFrameGroupBy, SeriesGroupBy))
325325

326+
is_ser_or_df = isinstance(obj, (ABCDataFrame, ABCSeries))
326327
context_manager: ContextManager
327328
if is_groupby:
328329
# When as_index=False, we combine all results using indices
@@ -341,8 +342,8 @@ def include_axis(colg) -> bool:
341342
if selected_obj.ndim == 1:
342343
for a in func:
343344
colg = obj._gotitem(selected_obj.name, ndim=1, subset=selected_obj)
344-
args = [self.axis, *self.args] if include_axis(colg) else self.args
345-
new_res = getattr(colg, op_name)(a, *args, **kwargs)
345+
this_args = [self.axis, *self.args] if is_ser_or_df else self.args
346+
new_res = getattr(colg, op_name)(a, *this_args, **kwargs)
346347
results.append(new_res)
347348

348349
# make sure we find a good name
@@ -353,8 +354,8 @@ def include_axis(colg) -> bool:
353354
indices = []
354355
for index, col in enumerate(selected_obj):
355356
colg = obj._gotitem(col, ndim=1, subset=selected_obj.iloc[:, index])
356-
args = [self.axis, *self.args] if include_axis(colg) else self.args
357-
new_res = getattr(colg, op_name)(func, *args, **kwargs)
357+
this_args = [self.axis, *self.args] if is_ser_or_df else self.args
358+
new_res = getattr(colg, op_name)(func, *this_args, **kwargs)
358359
results.append(new_res)
359360
indices.append(index)
360361
keys = selected_obj.columns.take(indices)

0 commit comments

Comments
 (0)