跟着老师写了一个竞技比赛模拟分析的代码,输入两个人的IPO(能力值)以及需要模拟的场次,就可以计算出双方的胜率.
from random import random
def printIntro():
print("这个程序模拟两个选手A和B的某种竞技比赛")
print("程序运行需要A和B的能力值(以0-1的小数表示)")
def getInputs():
a = eval(input("请输入选手A的能力值(0-1)"))
b = eval(input("请输入选手B的能力值(0-1)"))
n = eval(input("模拟比赛的场次:"))
return a, b, n
def gameOver(a, b):
return a == 15 or b == 15
def printSummary(winsA,winsB):
n = winsA + winsB
print("竞技分析开始, 共模拟{}场比赛".format(n))
print("选手A获胜{}场比赛, 占比{:0.1%}".format(winsA,winsA/n))
print("选手B获胜{}场比赛,占比{:0.1%}".format(winsB,winsB/n))
def simOneGame(probA,probB):
scoreA, scoreB = 0,0
serving = "A"
while not gameOver(scoreA,scoreB):
if serving == "A":
if random() < probA:
scoreA += 1
else:
serving = "B"
else:
if random() < probB:
scoreB += 1
else:
serving = "A"
return scoreA,scoreB
def simNGames(n,probA,probB):
winsA, winsB = 0,0
for i in range(n):
scoreA, scoreB = simOneGame(probA, probB)
if scoreA > scoreB:
winsA += 1
else:
winsB += 1
return winsA, winsB
def main():
printIntro()
probA, probB, n = getInputs()
winsA, winsB = simNGames(n, probA, probB)
printSummary(winsA, winsB)
main()