Skip to content

Commit 19ddf0d

Browse files
committed
added asv FillNA
1 parent 045e068 commit 19ddf0d

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
@@ -358,6 +358,21 @@ def time_category_size(self):
358358
self.draws.groupby(self.cats).size()
359359

360360

361+
class FillNA:
362+
def setup(self):
363+
N = 1500
364+
self.df = pd.DataFrame({
365+
'group' : [1] * N + [2] * N,
366+
'value' : [np.nan, 1.0] * N
367+
}).set_index('group')
368+
369+
def time_df_ffill(self):
370+
self.df.groupby('group').fillna(method='ffill')
371+
372+
def time_srs_ffill(self):
373+
self.df.groupby('group')['value'].fillna(method='ffill')
374+
375+
361376
class GroupByMethods:
362377

363378
param_names = ["dtype", "method", "application"]

0 commit comments

Comments
 (0)