Skip to content

Commit 7b2eaa4

Browse files
committed
Merge pull request #4207 from jreback/indexing_perf
PERF: add ix scalar get benchmark
2 parents 386d192 + 1bb9425 commit 7b2eaa4

File tree

1 file changed

+13
-10
lines changed

1 file changed

+13
-10
lines changed

vb_suite/indexing.py

+13-10
Original file line numberDiff line numberDiff line change
@@ -61,23 +61,26 @@
6161
bm_df_getitem2 = Benchmark(statement, setup,
6262
name='datamatrix_getitem_scalar')
6363

64-
setup = common_setup + """
65-
try:
66-
klass = DataMatrix
67-
except:
68-
klass = DataFrame
6964

65+
#----------------------------------------------------------------------
66+
# ix get scalar
67+
68+
setup = common_setup + """
7069
index = [tm.rands(10) for _ in xrange(1000)]
7170
columns = [tm.rands(10) for _ in xrange(30)]
72-
df = klass(np.random.rand(1000, 30), index=index,
71+
df = DataFrame(np.random.randn(1000, 30), index=index,
7372
columns=columns)
7473
idx = index[100]
7574
col = columns[10]
7675
"""
77-
statement = "df.get_value(idx, col)"
78-
bm_df_getitem3 = Benchmark(statement, setup,
79-
name='dataframe_get_value',
80-
start_date=datetime(2011, 11, 12))
76+
77+
indexing_frame_get_value_ix = Benchmark("df.ix[idx,col]", setup,
78+
name='indexing_frame_get_value_ix',
79+
start_date=datetime(2011, 11, 12))
80+
81+
indexing_frame_get_value = Benchmark("df.get_value(idx,col)", setup,
82+
name='indexing_frame_get_value',
83+
start_date=datetime(2011, 11, 12))
8184

8285
#----------------------------------------------------------------------
8386
# Boolean DataFrame row selection

0 commit comments

Comments
 (0)