LOL

reforge.py

import numpy as np
from numpy.random import uniform

def simulate(probability):

    n = 1
    while uniform() > probability:
        n += 1
    return n 
    
if __name__ == "__main__":

    np.random.seed(666)

    size = 10000
    probability = 0.1

    samples = np.array([simulate(probability) for _ in xrange(size)])
    
    print samples
    print samples.mean()
    print np.percentile(samples, 95)