From 1bb9425cae244c67715b82fdc5bf428cf3a48f13 Mon Sep 17 00:00:00 2001 From: jreback Date: Wed, 10 Jul 2013 20:35:20 -0400 Subject: [PATCH] PERF: add ix scalar get benchmark --- vb_suite/indexing.py | 23 +++++++++++++---------- 1 file changed, 13 insertions(+), 10 deletions(-) diff --git a/vb_suite/indexing.py b/vb_suite/indexing.py index 9f07cc6ed15c3..1264ae053ffca 100644 --- a/vb_suite/indexing.py +++ b/vb_suite/indexing.py @@ -61,23 +61,26 @@ bm_df_getitem2 = Benchmark(statement, setup, name='datamatrix_getitem_scalar') -setup = common_setup + """ -try: - klass = DataMatrix -except: - klass = DataFrame +#---------------------------------------------------------------------- +# ix get scalar + +setup = common_setup + """ index = [tm.rands(10) for _ in xrange(1000)] columns = [tm.rands(10) for _ in xrange(30)] -df = klass(np.random.rand(1000, 30), index=index, +df = DataFrame(np.random.randn(1000, 30), index=index, columns=columns) idx = index[100] col = columns[10] """ -statement = "df.get_value(idx, col)" -bm_df_getitem3 = Benchmark(statement, setup, - name='dataframe_get_value', - start_date=datetime(2011, 11, 12)) + +indexing_frame_get_value_ix = Benchmark("df.ix[idx,col]", setup, + name='indexing_frame_get_value_ix', + start_date=datetime(2011, 11, 12)) + +indexing_frame_get_value = Benchmark("df.get_value(idx,col)", setup, + name='indexing_frame_get_value', + start_date=datetime(2011, 11, 12)) #---------------------------------------------------------------------- # Boolean DataFrame row selection