You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.1 KiB
48 lines
1.1 KiB
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"):
|
|
"""
|
|
Generate an array of byte strings.
|
|
"""
|
|
retval = (
|
|
np.random.choice(RANDS_CHARS, size=nchars * np.prod(size))
|
|
.view((np.str_, nchars))
|
|
.reshape(size)
|
|
)
|
|
return retval.astype(dtype)
|
|
|
|
|
|
def randu_array(nchars, size, dtype="O"):
|
|
"""
|
|
Generate an array of unicode strings.
|
|
"""
|
|
retval = (
|
|
np.random.choice(RANDU_CHARS, size=nchars * np.prod(size))
|
|
.view((np.unicode_, nchars))
|
|
.reshape(size)
|
|
)
|
|
return retval.astype(dtype)
|
|
|
|
|
|
def rands(nchars):
|
|
"""
|
|
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))
|
|
|