Skip to content

Commit 0719f21

Browse files
committed
CLN: move set_grouper/get_grouper type pd.Grouper methods to internal
1 parent 98a4b37 commit 0719f21

File tree

3 files changed

+18
-18
lines changed

3 files changed

+18
-18
lines changed

pandas/core/groupby.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ def __init__(self, key=None, level=None, freq=None, axis=None, sort=False):
198198
def ax(self):
199199
return self.grouper
200200

201-
def get_grouper(self, obj):
201+
def _get_grouper(self, obj):
202202

203203
"""
204204
Parameters
@@ -210,10 +210,10 @@ def get_grouper(self, obj):
210210
a tuple of binner, grouper, obj (possibly sorted)
211211
"""
212212

213-
self.set_grouper(obj)
213+
self._set_grouper(obj)
214214
return self.binner, self.grouper, self.obj
215215

216-
def set_grouper(self, obj, sort=False):
216+
def _set_grouper(self, obj, sort=False):
217217
"""
218218
given an object and the specifcations, setup the internal grouper for this particular specification
219219
@@ -264,7 +264,7 @@ def set_grouper(self, obj, sort=False):
264264
self.grouper = ax
265265
return self.grouper
266266

267-
def get_binner_for_grouping(self, obj):
267+
def _get_binner_for_grouping(self, obj):
268268
raise NotImplementedError
269269

270270
@property
@@ -1697,7 +1697,7 @@ def __init__(self, index, grouper=None, obj=None, name=None, level=None,
16971697
elif isinstance(self.grouper, Grouper):
16981698

16991699
# get the new grouper
1700-
grouper = self.grouper.get_binner_for_grouping(self.obj)
1700+
grouper = self.grouper._get_binner_for_grouping(self.obj)
17011701
self.obj = self.grouper.obj
17021702
self.grouper = grouper
17031703
if self.name is None:
@@ -1807,7 +1807,7 @@ def _get_grouper(obj, key=None, axis=0, level=None, sort=True):
18071807

18081808
# a passed in Grouper, directly convert
18091809
if isinstance(key, Grouper):
1810-
binner, grouper, obj = key.get_grouper(obj)
1810+
binner, grouper, obj = key._get_grouper(obj)
18111811
return grouper, [], obj
18121812

18131813
# already have a BaseGrouper, just return it

pandas/tseries/resample.py

+10-10
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def __init__(self, freq='Min', closed=None, label=None, how='mean',
7676
super(TimeGrouper, self).__init__(freq=freq, axis=axis, **kwargs)
7777

7878
def resample(self, obj):
79-
self.set_grouper(obj, sort=True)
79+
self._set_grouper(obj, sort=True)
8080
ax = self.grouper
8181

8282
if isinstance(ax, DatetimeIndex):
@@ -93,7 +93,7 @@ def resample(self, obj):
9393
rs = self._resample_periods()
9494
else:
9595
obj = self.obj.to_timestamp(how=self.convention)
96-
self.set_grouper(obj)
96+
self._set_grouper(obj)
9797
rs = self._resample_timestamps()
9898
elif len(ax) == 0:
9999
return self.obj
@@ -104,11 +104,11 @@ def resample(self, obj):
104104
rs_axis.name = ax.name
105105
return rs
106106

107-
def get_grouper(self, obj):
108-
self.set_grouper(obj)
109-
return self.get_binner_for_resample()
107+
def _get_grouper(self, obj):
108+
self._set_grouper(obj)
109+
return self._get_binner_for_resample()
110110

111-
def get_binner_for_resample(self):
111+
def _get_binner_for_resample(self):
112112
# create the BinGrouper
113113
# assume that self.set_grouper(obj) has already been called
114114

@@ -121,12 +121,12 @@ def get_binner_for_resample(self):
121121
self.grouper = BinGrouper(bins, binlabels)
122122
return self.binner, self.grouper, self.obj
123123

124-
def get_binner_for_grouping(self, obj):
124+
def _get_binner_for_grouping(self, obj):
125125
# return an ordering of the transformed group labels,
126126
# suitable for multi-grouping, e.g the labels for
127127
# the resampled intervals
128-
ax = self.set_grouper(obj)
129-
self.get_binner_for_resample()
128+
ax = self._set_grouper(obj)
129+
self._get_binner_for_resample()
130130

131131
# create the grouper
132132
binner = self.binner
@@ -233,7 +233,7 @@ def _resample_timestamps(self):
233233
# assumes set_grouper(obj) already called
234234
axlabels = self.ax
235235

236-
self.get_binner_for_resample()
236+
self._get_binner_for_resample()
237237
grouper = self.grouper
238238
binner = self.binner
239239
obj = self.obj

pandas/tseries/tests/test_resample.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1134,7 +1134,7 @@ def test_apply_iteration(self):
11341134
df = DataFrame({'open': 1, 'close': 2}, index=ind)
11351135
tg = TimeGrouper('M')
11361136

1137-
_, grouper, _ = tg.get_grouper(df)
1137+
_, grouper, _ = tg._get_grouper(df)
11381138

11391139
# Errors
11401140
grouped = df.groupby(grouper, group_keys=False)
@@ -1151,7 +1151,7 @@ def test_panel_aggregation(self):
11511151
minor_axis=['A', 'B', 'C', 'D'])
11521152

11531153
tg = TimeGrouper('M', axis=1)
1154-
_, grouper, _ = tg.get_grouper(wp)
1154+
_, grouper, _ = tg._get_grouper(wp)
11551155
bingrouped = wp.groupby(grouper)
11561156
binagg = bingrouped.mean()
11571157

0 commit comments

Comments
 (0)