MatchAnalysis 竞技比赛模拟分析

跟着老师写了一个竞技比赛模拟分析的代码,输入两个人的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()