Numba random

A few numbaised useful functions for random sampling.

random_choice_numba(arr, prob)

Fast implementation of np.random.choice

Parameters:

Name	Type	Description	Default
arr			required
prob			required

Source code in june/utils/numba_random.py

@jit(nopython=True)
def random_choice_numba(arr, prob):
    """Fast implementation of np.random.choice

    Args:
        arr: 
        prob: 

    """
    return arr[np.searchsorted(np.cumsum(prob), random(), side="right")]

random_choice_numba_numpy(arr, prob, random_number)

Parameters:

Name	Type	Description	Default
arr			required
prob			required
random_number			required

Source code in june/utils/numba_random.py

@jit(nopython=True)
def random_choice_numba_numpy(arr, prob, random_number):
    """

    Args:
        arr: 
        prob: 
        random_number: 

    """

    return arr[np.searchsorted(np.cumsum(prob), random_number, side="right")]