python写一个验证蒙特霍尔问题的程序，输入测试次数，求出概率

这个问题属于计算问题，可以回答。以下是 Python 代码：

import random

def simulate_monty_hall(n):
    switch_win = 0
    stay_win = 0
    for i in range(n):
        doors = [0, 0, 1] # 三扇门，其中一扇有奖品
        random.shuffle(doors) # 随机打乱门的顺序
        first_choice = random.choice(doors) # 玩家随机选一扇门
        if first_choice == 1:
            switch_win += 1 # 如果玩家第一次就选了有奖品的门，换门就输了
        else:
            stay_win += 1 # 如果玩家第一次选了没奖品的门，留下门就输了
    switch_prob = switch_win / n # 换门获胜的概率
    stay_prob = stay_win / n # 留下门获胜的概率
    return switch_prob, stay_prob

n = int(input("请输入测试次数："))
switch_prob, stay_prob = simulate_monty_hall(n)
print("换门获胜的概率：", switch_prob)
print("留下门获胜的概率：", stay_prob)

这是一个蒙特霍尔问题的模拟程序，玩家需要在三扇门中选择一扇门，其中有一扇门有奖品，另外两扇门没有奖品。主持人会打开一扇没有奖品的门，让玩家判断是否换门。这个程序会模拟多次游戏，计算出换门和留下门获胜的概率。