参考文章:
https://zhuanlan.zhihu.com/p/262220753
做作业时遇到这个问题,在此记录一下。
原题为:某餐厅开展集齐十二生肖卡活动,顾客每消费一次,随机发给一张生肖卡。如果顾客集齐十二生肖卡,则可以免费消费一次。编写Python代码,应用蒙特卡罗方法(Monte Carlo Method)来估计平均消费多少次可以集齐一次十二生肖卡
我的解法:利用while循环,依次往数组内添加1到12之间的随机正整数(每个正整数对应一个生肖),统计不重复的正整数个数,若为12则集齐一次生肖卡,同时记录添加次数(即集齐一次消费多少次),最后将统计的次数放入另一个数组。在循环之外再估计平均消费多少次可以集齐一次十二生肖卡。编写代码得到结果为:大约消费37次即可集齐一次生肖卡。
完整代码如下:
import numpy as np
n = 3000 # 实验次数,可更改
record = [] # 记录每集齐一次生肖卡的消费次数
for i in range(0, n):
temp = 0 # 清零
credit = [] # 清零
while True:
credit.append(np.random.randint(0, 12)) # 收集一张卡片
num = np.unique(credit) # 求出不重复的卡片
temp = temp + 1 # 记录消费次数
if len(num) == 12: # 集齐12张则退出
break
record.append(temp) # 记录每集齐一次生肖卡的消费次数
print('平均一次实验需要消费' + str(sum(record) / n) + '次')