使用python解决三门问题

import random

'''
    python解决三门问题

    @author sage zsj

    三门问题(Monty Hall problem)亦称为蒙提霍尔问题、蒙特霍问题或蒙提霍尔悖论,
    大致出自美国的电视游戏节目Let's Make a Deal。问题名字来自该节目的主持人
    蒙提·霍尔(Monty Hall)。参赛者会看见三扇关闭了的门,其中一扇的后面有一辆汽车,
    选中后面有车的那扇门可赢得该汽车,另外两扇门后面则各藏有一只山羊。
    当参赛者选定了一扇门,但未去开启它的时候,节目主持人开启剩下两扇门的其中一扇,露出其中一只山羊。
    主持人其后会问参赛者要不要换另一扇仍然关上的门。问题是:换另一扇门会否增加参赛者赢得汽车的机率?
'''


class Door:
    contain = 'sheep'

    def __repr__(self):
        return 'There is a ' + self.contain + ' behind this door'


def three_door_problem(reselect, redo=100):
    right = 0
    for _ in range(redo):
        # 放置三道门,里面都是羊
        three_door = [Door(), Door(), Door()]
        # 随机将一道门后的羊换成车
        random.choice(three_door).contain = 'car'
        # 随便猜一道门
        choose_door = random.choice(three_door)
        # 打开另外两道门中含有羊的一道门
        rother_door = [door for door in three_door if not door is choose_door]
        for door in rother_door:
            if door.contain == 'sheep':
                rother_door.remove(door)
                break
        # 去掉错误答案后,是否重新选择另外一门
        if reselect == True:
            choose_door = rother_door.pop()
        # 统计选中车的次数
        if choose_door.contain == 'car':
            right += 1
    print('进行了{0}次三门问题,是否重选 = {1},选中车概率为{2:.2%}'.format(
        redo, reselect, right/redo))


three_door_problem(False, 10000)
three_door_problem(True, 10000)

运行结果

你可能感兴趣的:(python)