Skip to content

Commit 70c7b41

Browse files
authored
ASV: Add merge on categorical benchmarks (#45169)
1 parent ff0f8ba commit 70c7b41

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

asv_bench/benchmarks/join_merge.py

+12
Original file line numberDiff line numberDiff line change
@@ -262,12 +262,24 @@ def setup(self):
262262
Z=self.right_object["Z"].astype("category")
263263
)
264264

265+
self.left_cat_col = self.left_object.astype({"X": "category"})
266+
self.right_cat_col = self.right_object.astype({"X": "category"})
267+
268+
self.left_cat_idx = self.left_cat_col.set_index("X")
269+
self.right_cat_idx = self.right_cat_col.set_index("X")
270+
265271
def time_merge_object(self):
266272
merge(self.left_object, self.right_object, on="X")
267273

268274
def time_merge_cat(self):
269275
merge(self.left_cat, self.right_cat, on="X")
270276

277+
def time_merge_on_cat_col(self):
278+
merge(self.left_cat_col, self.right_cat_col, on="X")
279+
280+
def time_merge_on_cat_idx(self):
281+
merge(self.left_cat_idx, self.right_cat_idx, on="X")
282+
271283

272284
class MergeOrdered:
273285
def setup(self):

0 commit comments

Comments
 (0)