Skip to content

Commit c6cf9ee

Browse files
committed
Shrink benchmark
1 parent 71e1c4f commit c6cf9ee

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

asv_bench/benchmarks/groupby.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -501,15 +501,21 @@ class CumminMax:
501501
]
502502

503503
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+
)
506511
null_vals = vals.astype(float, copy=True)
507512
null_vals[::2, :] = np.nan
508513
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
513519
self.df = df
514520
self.null_df = null_df
515521

0 commit comments

Comments
 (0)