Skip to content

Commit 5f9478f

Browse files
committed
add ASV
1 parent 006048c commit 5f9478f

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

asv_bench/benchmarks/groupby.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -369,6 +369,21 @@ def time_category_size(self):
369369
self.draws.groupby(self.cats).size()
370370

371371

372+
class Shift:
373+
def setup(self):
374+
N = 18
375+
self.df = DataFrame({
376+
'g' : ['a', 'b'] * 9,
377+
'v' : list(range(N))
378+
})
379+
380+
def time_defaults(self):
381+
self.df.groupby('g').shift()
382+
383+
def time_fill_value(self):
384+
self.df.groupby('g').shift(fill_value=99)
385+
386+
372387
class FillNA:
373388
def setup(self):
374389
N = 100

0 commit comments

Comments
 (0)