File tree Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Original file line number Diff line number Diff line change @@ -223,11 +223,16 @@ class CategoricalSlicing(object):
223223
224224 def setup (self , index ):
225225 N = 10 ** 6
226- values = list ('a' * N + 'b' * N + 'c' * N )
226+ categories = ['a' , 'b' , 'c' ]
227+ values = [0 ] * N + [1 ] * N + [2 ] * N
227228 indices = {
228- 'monotonic_incr' : pd .Categorical (values ),
229- 'monotonic_decr' : pd .Categorical (reversed (values )),
230- 'non_monotonic' : pd .Categorical (list ('abc' * N ))}
229+ 'monotonic_incr' : pd .Categorical .from_codes (values ,
230+ categories = categories ),
231+ 'monotonic_decr' : pd .Categorical .from_codes (list (reversed (values )),
232+ categories = categories ),
233+ 'non_monotonic' : pd .Categorical .from_codes ([0 , 1 , 2 ] * N ,
234+ categories = categories )
235+ }
231236 self .data = indices [index ]
232237
233238 self .scalar = 10000
You can’t perform that action at this time.
0 commit comments