Skip to content
Prev Previous commit
asv_bench: fix indentation and simplify
  • Loading branch information
erikcs committed Dec 11, 2016
commit df1edf8e182047f330df02e1697ad6feaff1dbf4
27 changes: 12 additions & 15 deletions asv_bench/benchmarks/reshape.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,21 +80,18 @@ class wide_to_long_big(object):
goal_time = 0.2

def setup(self):
vars = 'ABCD'
nyrs = 20
nidvars = 20
N = 5000
yrvars = []
for var in vars:
for yr in range(1, nyrs + 1):
yrvars.append(var + str(yr))

yearobs = dict(zip(yrvars, np.random.randn(len(yrvars), N)))
idobs = dict(zip(range(nidvars), np.random.rand(nidvars, N)))

self.df = pd.concat([pd.DataFrame(idobs), pd.DataFrame(yearobs)],
axis=1)
self.vars = vars
vars = 'ABCD'
nyrs = 20
nidvars = 20
N = 5000
yrvars = []
for var in vars:
for yr in range(1, nyrs + 1):
yrvars.append(var + str(yr))

self.df = pd.DataFrame(np.random.randn(N, nidvars + len(yrvars)),
columns=list(range(nidvars)) + yrvars)
self.vars = vars

def time_wide_to_long_big(self):
self.df['id'] = self.df.index
Expand Down