Skip to content

Commit 2abc9a9

Browse files
jbrockmendelfeefladder
authored andcommitted
BENCH: tighter targeting on ChainedIndexing benchmark (pandas-dev#43317)
1 parent c4a38e5 commit 2abc9a9

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

asv_bench/benchmarks/indexing.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -399,12 +399,14 @@ class ChainIndexing:
399399

400400
def setup(self, mode):
401401
self.N = 1000000
402+
self.df = DataFrame({"A": np.arange(self.N), "B": "foo"})
402403

403404
def time_chained_indexing(self, mode):
405+
df = self.df
406+
N = self.N
404407
with warnings.catch_warnings(record=True):
405408
with option_context("mode.chained_assignment", mode):
406-
df = DataFrame({"A": np.arange(self.N), "B": "foo"})
407-
df2 = df[df.A > self.N // 2]
409+
df2 = df[df.A > N // 2]
408410
df2["C"] = 1.0
409411

410412

0 commit comments

Comments
 (0)