From 8d36f2bd59074ff1d5a90c0690ffef9baa08a5ac Mon Sep 17 00:00:00 2001 From: phofl Date: Tue, 28 Dec 2021 20:51:54 +0100 Subject: [PATCH] ASV: Add benchmarks for isin --- asv_bench/benchmarks/algos/isin.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/asv_bench/benchmarks/algos/isin.py b/asv_bench/benchmarks/algos/isin.py index 5d7a76bc01d49..37fa0b490bd9e 100644 --- a/asv_bench/benchmarks/algos/isin.py +++ b/asv_bench/benchmarks/algos/isin.py @@ -2,6 +2,7 @@ from pandas import ( Categorical, + Index, NaT, Series, date_range, @@ -326,3 +327,16 @@ def setup(self): def time_isin(self): self.series.isin(self.values) + + +class IsInIndexes: + def setup(self): + self.range_idx = Index(range(1000)) + self.index = Index(list(range(1000))) + self.series = Series(np.random.randint(100_000, size=1000)) + + def time_isin_range_index(self): + self.series.isin(self.range_idx) + + def time_isin_index(self): + self.series.isin(self.index)