diff --git a/asv_bench/benchmarks/reshape.py b/asv_bench/benchmarks/reshape.py index f29b2fc663eed..c83cd9a925f6d 100644 --- a/asv_bench/benchmarks/reshape.py +++ b/asv_bench/benchmarks/reshape.py @@ -102,6 +102,7 @@ def setup(self, dtype): columns = np.arange(n) if dtype == "int": values = np.arange(m * m * n).reshape(m * m, n) + self.df = DataFrame(values, index, columns) else: # the category branch is ~20x slower than int. So we # cut down the size a bit. Now it's only ~3x slower. @@ -111,7 +112,10 @@ def setup(self, dtype): values = np.take(list(string.ascii_letters), indices) values = [pd.Categorical(v) for v in values.T] - self.df = DataFrame({i: cat for i, cat in enumerate(values)}, index, columns) + self.df = DataFrame( + {i: cat for i, cat in enumerate(values)}, index, columns + ) + self.df2 = self.df.iloc[:-1] def time_full_product(self, dtype):