Python蒙特卡洛算法解十二生肖卡问题

参考文章：

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) + '次')