Monte Carlo for COMP 424

montecarlo.py

from random import random, shuffle

def monte_carlo(draws, reds, blues):
    cases = []
    for i in xrange(draws):
        red_draws = [1] * reds
        blue_draws = [0] * blues
        total = red_draws + blue_draws
        shuffle(total)
        utility = total.pop() + total.pop() # pop two off
        cases.append(utility)

    return sum(cases)/float(draws)

print "3 red and 2 blue: " + str(monte_carlo(10000, 3, 2))
print "30 red and 20 blue: " + str(monte_carlo(10000, 30, 20))