monty hall

monty_hall.py

import random

ITERS = 10000
a = ['g', 'g', 'c']

stay_results = []
for i in range(ITERS):
	random.shuffle(a)
	init_choice = int(random.random()*len(a))
	goat = [i for i in range(3) if i != init_choice and a[i] != 'c'][0]

	if a[init_choice] == 'c':
		stay_results.append(1)
	else:
		stay_results.append(0)


switch_results = []
for i in range(ITERS):
	random.shuffle(a)
	init_choice = int(random.random()*len(a))
	goat = [i for i in range(3) if i != init_choice and a[i] != 'c'][0]
	switch = list(set([0,1,2]) - set([init_choice, goat]))[0]

	if a[switch] == 'c':
		switch_results.append(1)
	else:
		switch_results.append(0)

print("SWITCH: {}".format(sum(switch_results)/ITERS))
print("STAY: {}".format(sum(stay_results)/ITERS))