python实现三开门问题（蒙特卡罗）

import numpy as np

n =5000  # 迭代次数

x = 0  # 不改主意的次数
y = 0  # 改主意的次数

a = np.array([1, 2, 3])  #奖品的门
b = np.array([1, 2, 3])  # 奖品车所在的门

for i in range(n):

    c = np.random.choice(a, 1)  # 参加人所选的门

    d = np.random.choice(b, 1)  # 模拟车在的位置 在123门里有随机一个是车

    if c == d:  # 如果一开始人选的就是车的位置
        x = x + 1

    else:  # 如果不是
        y = y + 1

p = x / n  # 这是不改主意中奖的概率
q = y / n  # 这是改主意中奖的概率

print(x)
print(y)
print(p)
print(q)