Python小猫掉进山洞问题

Question

一只小猫不幸失明，更加不幸的是这只小猫掉进一个山洞，这个山洞有三个门，第一个进去之后2小时可以回到地面，第二个门进去之后4小时回到原始出发点，第三个门进去之后6小时回到原始出发点。由于小猫失明，每次都只能随机地选其中的一个门走，那么可怜的小猫走出山洞的期望时间是多少？提示：模拟小猫走到地面的过程10000次，并记录每次走到地面消耗的时间，求平均值（模拟次数越多，该值越接近于期望时间）

Answer 1

import random

doors = ['d1', 'd2', 'd3']
ans = []
n = 0

while n < 10000:
    if random.choice(doors) == 'd1':
        ans.append(2)
        n += 1
    if random.choice(doors) == 'd2':
        ans.append(4)
    if random.choice(doors) == 'd3':
        ans.append(6)
    
print(sum(ans) / 10000)     # 平均在12左右