stochastic_acceptance.py

import random


def sample(freqs_or_probs, precomputed_max_over_freqs_or_probs=None):
    x = freqs_or_probs
    n = len(x)
    max_value = precomputed_max_over_freqs_or_probs or max(x)
    
    while True:
        i = int(n * random.random())
        if random.random() < x[i] / max_value:
            return i