차,염소,염소

monty-hall-problem.py

import random

def simul1():
    rooms = [0, 0, 1]
    random.shuffle(rooms)
    i = random.randrange(3) # initial

    while True:
        j = random.randrange(3) # goat
        if j != i and rooms[j] == 0:
            break

    k = ({0, 1, 2} - {i, j}).pop()
    return rooms[k] == 1 # switch

tries = 100000
wins = sum(simul1() for i in range(tries))
print(float(wins) / tries)