auto_monty.py

#!/bin/python3

import random

doors = ['car', 'goat', 'goat']

def monty(switch_doors):
    random.shuffle(doors)
    number = random.randint(0, 2)
    remaining = [0, 1, 2]
    remaining.remove(number)
    random.shuffle(remaining)
    preview = next(x for x in remaining if doors[x] == 'goat')

    remaining.remove(preview)
    remaining = remaining[0]

    if switch_doors:
        number, remaining = remaining, number
    return doors[number] == 'car'


def run_test(switch, iterations):
    win = 0
    loss = 0
    for _ in range(iterations):
        if monty(switch):
            win += 1
        else:
            loss += 1
    return win, loss


def main():
    iterations = int(input('iterations? '))

    win, loss = run_test(False, iterations)
    print('no switch:\twins=' + str(win) + '\tloss=' + str(loss))

    win, loss = run_test(True, iterations)
    print('switch:\t\twins=' + str(win) + '\tloss=' + str(loss))


if __name__ == '__main__':
    main()