炉石传说上传说需要的局数~Python日常~

如果已知自己的胜率，那么从五级0星开始，到上传说大概需要多少局？
这次使用暴力的算法（尝试10000次计算，求平均值）

需要用到的基础知识：

最简单的随机数的产生方法：
需要import的库：
import random

产生一个随机浮点数[0,1]
random.uniform(0, 1)

产生一个随机整数[0,100]
random.randint(0, 100)

基本思路：

输入容错：

1. 用户输入自己胜率winrate
   如果winrate小于50%，是不可能上传说的，直接退出
   如果winrate大于100%，则认为输入有误，不进行计算

实际计算：

1. 随机产生一个[0, 1]的浮点数，如果这个数大于胜率winrate，则输，否则为胜
2. 计算净胜场，赢一局+1，输一局-1
3. 如果净胜场>30（简单算了一下从五级0星到传说貌似需要30胜。。。）
4. 将上步骤重复多次，最后算出平均需要的局数
   (忽略了掉到5级以下可能出现的连胜)
   具体代码如下:

import random
import math

#产生一个随机浮点数[0,1]
#print(random.uniform(0,1))
#产生一个随机整数[0,100]
#print(random.randint(0,100))

#暴力计算的次数
COMPUTE_NUM = 10000
#需要净胜利的局数
NEED_TO_WIN = 30

print("输入你的胜率~")
winrate = float(input())
if winrate <= 0.5:
    print("胜率这么低还想上传说？早点洗洗睡吧")
elif winrate > 1:
    print("输入胜率有误")
else:
    print("你的胜率为: " + str(winrate))

#存储需要的胜利局数
result = []
#本次计算的净胜场
win_count = 0
#本次计算实际进行的局数
game_count = 0

win_total = 0
#重复很多次来减小误差
for each in range(COMPUTE_NUM):
    while win_count < NEED_TO_WIN:
        rnd = random.uniform(0, 1)
        #print(rnd)
        #print(winrate)
        if rnd <= winrate:
            #胜利~
            #print("win")
            win_count += 1
        else:
            #print("lose")
            win_count -= 1

        game_count += 1

    result.append(game_count)
    win_total += game_count
    #print(win_total)
    win_count = 0
    game_count = 0

#print(win_total)
average = win_total/COMPUTE_NUM
#标准差SD = sqrt((1 / n) * Σ(x - average)^2)
temp_total = 0
for each in result:
    temp_total += (each - average)**2
sd = math.sqrt(temp_total / COMPUTE_NUM)
print("上传说，你平均需要%f局比赛"%average)
print("标准差为:%f"%sd)