@@ -63,17 +63,17 @@ cdef class _BaseGrouper:
63
63
)
64
64
return cached_index, cached_series
65
65
66
- cdef inline _update_cached_objs(self , object cached_typ , object cached_ityp ,
66
+ cdef inline _update_cached_objs(self , object cached_series , object cached_index ,
67
67
Slider islider, Slider vslider):
68
68
# See the comment in indexes/base.py about _index_data.
69
69
# We need this for EA-backed indexes that have a reference
70
70
# to a 1-d ndarray like datetime / timedelta / period.
71
- cached_ityp ._engine.clear_mapping()
72
- cached_ityp ._cache.clear() # e.g. inferred_freq must go
73
- cached_typ ._mgr.set_values(vslider.buf)
71
+ cached_index ._engine.clear_mapping()
72
+ cached_index ._cache.clear() # e.g. inferred_freq must go
73
+ cached_series ._mgr.set_values(vslider.buf)
74
74
75
75
cdef inline object _apply_to_group(self ,
76
- object cached_typ , object cached_ityp ,
76
+ object cached_series , object cached_index ,
77
77
bint initialized):
78
78
"""
79
79
Call self.f on our new group, then update to the next group.
@@ -83,7 +83,7 @@ cdef class _BaseGrouper:
83
83
84
84
# NB: we assume that _update_cached_objs has already cleared cleared
85
85
# the cache and engine mapping
86
- res = self .f(cached_typ )
86
+ res = self .f(cached_series )
87
87
res = extract_result(res)
88
88
if not initialized:
89
89
# On the first pass, we check the output shape to see
@@ -140,7 +140,7 @@ cdef class SeriesBinGrouper(_BaseGrouper):
140
140
object res
141
141
bint initialized = 0
142
142
Slider vslider, islider
143
- object cached_typ = None , cached_ityp = None
143
+ object cached_series = None , cached_index = None
144
144
145
145
counts = np.zeros(self .ngroups, dtype = np.int64)
146
146
@@ -160,7 +160,9 @@ cdef class SeriesBinGrouper(_BaseGrouper):
160
160
161
161
result = np.empty(self .ngroups, dtype = ' O' )
162
162
163
- cached_ityp, cached_typ = self ._init_dummy_series_and_index(islider, vslider)
163
+ cached_index, cached_series = self ._init_dummy_series_and_index(
164
+ islider, vslider
165
+ )
164
166
165
167
start = 0
166
168
try :
@@ -172,9 +174,9 @@ cdef class SeriesBinGrouper(_BaseGrouper):
172
174
vslider.move(start, end)
173
175
174
176
self ._update_cached_objs(
175
- cached_typ, cached_ityp , islider, vslider)
177
+ cached_series, cached_index , islider, vslider)
176
178
177
- res, initialized = self ._apply_to_group(cached_typ, cached_ityp ,
179
+ res, initialized = self ._apply_to_group(cached_series, cached_index ,
178
180
initialized)
179
181
start += group_size
180
182
@@ -236,7 +238,7 @@ cdef class SeriesGrouper(_BaseGrouper):
236
238
object res
237
239
bint initialized = 0
238
240
Slider vslider, islider
239
- object cached_typ = None , cached_ityp = None
241
+ object cached_series = None , cached_index = None
240
242
241
243
labels = self .labels
242
244
counts = np.zeros(self .ngroups, dtype = np.int64)
@@ -248,7 +250,9 @@ cdef class SeriesGrouper(_BaseGrouper):
248
250
249
251
result = np.empty(self .ngroups, dtype = ' O' )
250
252
251
- cached_ityp, cached_typ = self ._init_dummy_series_and_index(islider, vslider)
253
+ cached_index, cached_series = self ._init_dummy_series_and_index(
254
+ islider, vslider
255
+ )
252
256
253
257
start = 0
254
258
try :
@@ -268,9 +272,9 @@ cdef class SeriesGrouper(_BaseGrouper):
268
272
vslider.move(start, end)
269
273
270
274
self ._update_cached_objs(
271
- cached_typ, cached_ityp , islider, vslider)
275
+ cached_series, cached_index , islider, vslider)
272
276
273
- res, initialized = self ._apply_to_group(cached_typ, cached_ityp ,
277
+ res, initialized = self ._apply_to_group(cached_series, cached_index ,
274
278
initialized)
275
279
276
280
start += group_size
@@ -293,20 +297,20 @@ cdef class SeriesGrouper(_BaseGrouper):
293
297
return result, counts
294
298
295
299
296
- cpdef inline extract_result(object res, bint squeeze = True ):
300
+ cpdef inline extract_result(object res):
297
301
""" extract the result object, it might be a 0-dim ndarray
298
302
or a len-1 0-dim, or a scalar """
299
303
if hasattr (res, " _values" ):
300
304
# Preserve EA
301
305
res = res._values
302
- if squeeze and res.ndim == 1 and len (res) == 1 :
306
+ if res.ndim == 1 and len (res) == 1 :
303
307
res = res[0 ]
304
308
if hasattr (res, ' values' ) and is_array(res.values):
305
309
res = res.values
306
310
if is_array(res):
307
311
if res.ndim == 0 :
308
312
res = res.item()
309
- elif squeeze and res.ndim == 1 and len (res) == 1 :
313
+ elif res.ndim == 1 and len (res) == 1 :
310
314
res = res[0 ]
311
315
return res
312
316
0 commit comments