Skip to content

Commit ea5d08d

Browse files
committed
CLN: update asv benchmarks to reflect astype categorical deprecation
1 parent 5916788 commit ea5d08d

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

asv_bench/benchmarks/categoricals.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -132,16 +132,18 @@ def setup(self):
132132
ncats = 100
133133

134134
self.s_str = pd.Series(tm.makeCategoricalIndex(N, ncats)).astype(str)
135-
self.s_str_cat = self.s_str.astype('category')
135+
self.s_str_cat = pd.Series(self.s_str, dtype='category')
136136
with warnings.catch_warnings(record=True):
137-
self.s_str_cat_ordered = self.s_str.astype('category',
138-
ordered=True)
137+
str_cat_type = pd.CategoricalDtype(set(self.s_str),
138+
ordered=True)
139+
self.s_str_cat_ordered = self.s_str.astype(str_cat_type)
139140

140141
self.s_int = pd.Series(np.random.randint(0, ncats, size=N))
141-
self.s_int_cat = self.s_int.astype('category')
142+
self.s_int_cat = pd.Series(self.s_int, dtype='category')
142143
with warnings.catch_warnings(record=True):
143-
self.s_int_cat_ordered = self.s_int.astype('category',
144-
ordered=True)
144+
int_cat_type = pd.Categorical(set(self.s_int),
145+
ordered=True)
146+
self.s_int_cat_ordered = self.s_int.astype(int_cat_type)
145147

146148
def time_rank_string(self):
147149
self.s_str.rank()

0 commit comments

Comments
 (0)