比赛模拟
代码:
from random import random
def printIntro():
print("学号:2020310143040")
print("====乒乓球竞技分析====")
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("模拟比赛的场次: "))
raceType = eval(input("比赛类型:1为单打,2为双打 "))
return a, b, n, raceType
def simNGames(n, probA, probB, raceType):
winsA, winsB = 0, 0
if raceType == 1:
for i in range(n):
for j in range(7):
scoreA, scoreB = simOneGame(probA, probB)
if scoreA > scoreB:
winsA += 1
else:
winsB += 1
return winsA, winsB
else:
for i in range(n):
for j in range(5):
scoreA, scoreB = simOneGame(probA, probB)
if scoreA > scoreB:
winsA += 1
else:
winsB += 1
return winsA, winsB
def gameOver(a,b):
return a==11 or b==11
def gameOver2(a,b):
return a==12 or b==12
def simOneGame(probA, probB):
scoreA, scoreB = 0, 0
serving = "A"
while not gameOver(scoreA, scoreB):
if scoreA==10 and scoreB==10:
return(simtwoGame2(probA,probB))
if serving == "A":
if random() < probA:
scoreA += 1
else:
serving="B"
else:
if random() < probB:
scoreB += 1
else:
serving="A"
return scoreA, scoreB
def simtwoGame2(probA,probB):
scoreA,scoreB=10,10
serving = "A"
while not gameOver2(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 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 main():
printIntro()
probA, probB, n ,raceType= getInputs()
winsA, winsB = simNGames(n, probA, probB, raceType)
printSummary(winsA, winsB)
main()
运行结果:

浙公网安备 33010602011771号