File tree Expand file tree Collapse file tree 1 file changed +12
-6
lines changed Expand file tree Collapse file tree 1 file changed +12
-6
lines changed Original file line number Diff line number Diff line change @@ -501,15 +501,21 @@ class CumminMax:
501
501
]
502
502
503
503
def setup (self , dtype , method ):
504
- N = 1_000_000
505
- vals = np .random .randint (0 , 1000 , (N , 10 ))
504
+ N = 500_000
505
+ rg = np .random .default_rng (seed = 0 )
506
+ vals = rg .integers (
507
+ 0 ,
508
+ 1000 ,
509
+ (N , 5 ),
510
+ )
506
511
null_vals = vals .astype (float , copy = True )
507
512
null_vals [::2 , :] = np .nan
508
513
null_vals [::3 , :] = np .nan
509
- df = DataFrame (vals , columns = list ("abcdefghij" ), dtype = dtype )
510
- null_df = DataFrame (null_vals , columns = list ("abcdefghij" ), dtype = dtype )
511
- df ["key" ] = np .random .randint (0 , 100 , size = N )
512
- null_df ["key" ] = np .random .randint (0 , 100 , size = N )
514
+ df = DataFrame (vals , columns = list ("abcde" ), dtype = dtype )
515
+ null_df = DataFrame (null_vals , columns = list ("abcde" ), dtype = dtype )
516
+ keys = rg .integers (0 , 100 , size = N )
517
+ df ["key" ] = keys
518
+ null_df ["key" ] = keys
513
519
self .df = df
514
520
self .null_df = null_df
515
521
You can’t perform that action at this time.
0 commit comments