Skip to content

Commit ef2340f

Browse files
committed
TST: asv for DataFrame from dict with index/columns
1 parent ce8f000 commit ef2340f

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

asv_bench/benchmarks/frame_ctor.py

+12-6
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,11 @@ class FromDicts(object):
1616

1717
def setup(self):
1818
N, K = 5000, 50
19-
index = tm.makeStringIndex(N)
20-
columns = tm.makeStringIndex(K)
21-
frame = DataFrame(np.random.randn(N, K), index=index, columns=columns)
19+
self.index = tm.makeStringIndex(N)
20+
self.columns = tm.makeStringIndex(K)
21+
frame = DataFrame(np.random.randn(N, K), index=self.index,
22+
columns=self.columns)
2223
self.data = frame.to_dict()
23-
self.some_dict = list(self.data.values())[0]
2424
self.dict_list = frame.to_dict(orient='records')
2525
self.data2 = {i: {j: float(j) for j in range(100)}
2626
for i in range(2000)}
@@ -31,8 +31,14 @@ def time_list_of_dict(self):
3131
def time_nested_dict(self):
3232
DataFrame(self.data)
3333

34-
def time_dict(self):
35-
Series(self.some_dict)
34+
def time_nested_dict_index(self):
35+
DataFrame(self.data, index=self.index)
36+
37+
def time_nested_dict_columns(self):
38+
DataFrame(self.data, columns=self.columns)
39+
40+
def time_nested_dict_index_columns(self):
41+
DataFrame(self.data, index=self.index, columns=self.columns)
3642

3743
def time_nested_dict_int64(self):
3844
# nested dict, integer indexes, regression described in #621

0 commit comments

Comments
 (0)