37 lines
858 B
Python
37 lines
858 B
Python
|
import string
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
|
||
|
|
def randbool(size=(), p: float = 0.5):
|
||
|
|
return np.random.rand(*size) <= p
|
||
|
|
|
||
|
|
|
||
|
|
RANDS_CHARS = np.array(list(string.ascii_letters + string.digits), dtype=(np.str_, 1))
|
||
|
|
RANDU_CHARS = np.array(
|
||
|
|
list("".join(map(chr, range(1488, 1488 + 26))) + string.digits),
|
||
|
|
dtype=(np.unicode_, 1),
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def rands_array(nchars, size, dtype="O", replace=True) -> np.ndarray:
|
||
|
|
"""
|
||
|
|
Generate an array of byte strings.
|
||
|
|
"""
|
||
|
|
retval = (
|
||
|
|
np.random.choice(RANDS_CHARS, size=nchars * np.prod(size), replace=replace)
|
||
|
|
.view((np.str_, nchars))
|
||
|
|
.reshape(size)
|
||
|
|
)
|
||
|
|
return retval.astype(dtype)
|
||
|
|
|
||
|
|
|
||
|
|
def rands(nchars) -> str:
|
||
|
|
"""
|
||
|
|
Generate one random byte string.
|
||
|
|
|
||
|
|
See `rands_array` if you want to create an array of random strings.
|
||
|
|
|
||
|
|
"""
|
||
|
|
return "".join(np.random.choice(RANDS_CHARS, nchars))
|