Python解决羊车门问题

有3扇关闭的门，一扇门后面停着汽车，其余门后是山羊，只有主持人知道每扇门后面是什么。参赛者可以选择一扇门，在开启它之前，主持人会开启另外一扇门，露出门后的山羊，然后允许参赛者更换自己的选择。请问：参赛者更换选择后能否增加猜中汽车的机会？

import random

def monty_hall_simulation(num_trials):
    stay_wins = 0
    switch_wins = 0

    for _ in range(num_trials):
        # 三扇门，一辆汽车和两只山羊随机放置
        doors = ['car', 'goat', 'goat']
        random.shuffle(doors)

        # 参赛者首先选择一扇门
        participant_choice = random.randint(0, 2)

        # 主持人打开一扇门，露出山羊
        doors_without_participant_choice = [i for i in range(3)
                                            if i != participant_choice and doors[i] == 'goat']
        door_opened_by_host = random.choice(doors_without_participant_choice)

        # 参赛者决定是否更换选择
        # 将符合条件生成列表，选择第一个元素
        switch_choice = [i for i in range(3)
                         if i != participant_choice and i != door_opened_by_host][0]
        print(switch_choice)
        # 统计结果
        if doors[participant_choice] == 'car':
            stay_wins += 1
        elif doors[switch_choice] == 'car':
            switch_wins += 1

    # 计算概率
    stay_probability = stay_wins / num_trials
    switch_probability = switch_wins / num_trials

    return stay_probability, switch_probability


# 进行模拟，例如10000次
num_trials = eval(input('请输入要进行测试的次数：'))
stay_probability, switch_probability = monty_hall_simulation(num_trials)

# 打印结果
print(f"不更换选择时的获胜概率：{stay_probability:.2f}")
print(f"更换选择时的获胜概率：{switch_probability:.2f}")