Code to run simulation of the Monty hall problem

MontyPython.py

# -*- coding: utf-8 -*-
import random

def round(switch=True):
  doors = ['car', 'goat', 'goat']
  pick = random.randrange(3)
  print "the RNG picked door %s" % (pick + 1)
  #monty knows where the car is
  car = doors.index('car')
  #and he opens a door without the car
  monty_shows = random.choice([i for i in range(3) if i not in [pick, car]])
  print "Monty opens door %s and reveals a goat!" % (monty_shows + 1)
  if switch:
    pick = [i for i in range(3) if i not in [pick, monty_shows]][0]
    print "the RNG switched to door %s" % (pick + 1)
  if pick== car:
    return 1
  else:
    return 0

def simulation(handle, generations=1000):
  switch = 0
  stay = 0
  for gen in range(generations):
    switch += round(switch=True)
    stay   += round(switch= False)
    handle.write("%s,%s\n" % (switch/(gen + 1.0), stay/(gen + 1.0)))