Skip to content

Commit e4cdd07

Browse files
committed
DOC: Add comments for the single level groupby
1 parent ebf8bc3 commit e4cdd07

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

pandas/core/groupby.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -2586,14 +2586,17 @@ def _get_grouper(obj, key=None, axis=0, level=None, sort=True,
25862586
"""
25872587
group_axis = obj._get_axis(axis)
25882588

2589-
# validate that the passed level is compatible with the passed
2589+
# validate that the passed single level is compatible with the passed
25902590
# axis of the object
25912591
if level is not None:
2592+
# TODO: These two conditions are almost same.
2593+
# We should combine two.
25922594
if isinstance(group_axis, MultiIndex):
25932595
if is_list_like(level) and len(level) == 1:
25942596
level = level[0]
25952597

25962598
if key is None and is_scalar(level):
2599+
# Get the level values from group_axis
25972600
key = group_axis.get_level_values(level)
25982601
level = None
25992602

@@ -2619,6 +2622,8 @@ def _get_grouper(obj, key=None, axis=0, level=None, sort=True,
26192622
raise ValueError('level > 0 or level < -1 only valid with '
26202623
' MultiIndex')
26212624

2625+
# NOTE: `group_axis` and `group_axis.get_level_values(level)`
2626+
# are same in this section.
26222627
level = None
26232628
key = group_axis
26242629

0 commit comments

Comments
 (0)