From b5b302245548a849d903eb0fbe460fbce9e4c896 Mon Sep 17 00:00:00 2001 From: Luke Manley Date: Sun, 2 Jan 2022 22:01:59 -0500 Subject: [PATCH] ASV: Add merge on categorical benchmarks --- asv_bench/benchmarks/join_merge.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/asv_bench/benchmarks/join_merge.py b/asv_bench/benchmarks/join_merge.py index 27eaecff09d0f..ad40adc75c567 100644 --- a/asv_bench/benchmarks/join_merge.py +++ b/asv_bench/benchmarks/join_merge.py @@ -262,12 +262,24 @@ def setup(self): Z=self.right_object["Z"].astype("category") ) + self.left_cat_col = self.left_object.astype({"X": "category"}) + self.right_cat_col = self.right_object.astype({"X": "category"}) + + self.left_cat_idx = self.left_cat_col.set_index("X") + self.right_cat_idx = self.right_cat_col.set_index("X") + def time_merge_object(self): merge(self.left_object, self.right_object, on="X") def time_merge_cat(self): merge(self.left_cat, self.right_cat, on="X") + def time_merge_on_cat_col(self): + merge(self.left_cat_col, self.right_cat_col, on="X") + + def time_merge_on_cat_idx(self): + merge(self.left_cat_idx, self.right_cat_idx, on="X") + class MergeOrdered: def setup(self):