Resampling wheel (weighted random sampling) in python

resampler.py

def sample(weights, size=1):
    beta = 0
    samples = []
    w_max = max(weights)
    index = random.randint(1, N)
    while len(samples) < size:
        beta += random.uniform(0, 2 * w_max)
        while w[index] < beta:
            beta -= weights[index]
            index += 1
        samples.append(index)
    return samples