monty hall problem with monte carlo

gistfile1.txt

#!/usr/bin/python3

import random as r


def random_doors():
    doors = [False] * 3
    doors[r.randint(0, 2)] = True
    return doors


def choose_door():
    return r.randint(0, 2)


def reveal_door(doors, not_door):
    for i in range(len(doors)):
        if not doors[i] and i != not_door:
            doors[i] = None
            break

    return doors


def choose_remaining_door(doors, door):
    for i in range(len(doors)):
        if doors[i] is not None and i != door:
            return i


def run_experiment(switch):
    win_count = 0
    total = 100000
    for i in range(total):
        doors = random_doors()
        door = choose_door()
        if switch:
            doors = reveal_door(doors, door)
            door = choose_remaining_door(doors, door)

        if doors[door]:
            win_count += 1

    print("win chance when switch=%s: %f" % (switch, win_count / total))


run_experiment(switch=False)
run_experiment(switch=True)

# prints
# win chance when switch=False: 0.334480
# win chance when switch=True: 0.666830