问带有锦标赛功能的CS50实验室6 world_cup问题

EN

# Simulate a sports tournament

import csv
import sys
import random
from collections import defaultdict

# Number of simluations to run
N = 1000


def main():

    # Ensure correct usage
    if len(sys.argv) != 2:
        sys.exit("Usage: python tournament.py FILENAME")

    teams = []
    # TODO: Read teams into memory from file
    file = sys.argv[1]
    with open(file,"r") as fil:
        reader = csv.DictReader(fil)

        for row in reader:
            name = row["team"]
            rating = int(row["rating"])
            teams += [name , rating]
            #x = [name , rating]
            #teams.append(x)

    counts = {}
    # TODO: Simulate N tournaments and keep track of win counts
    for i in range(N):
        teamname = simulate_tournament(teams)
        #counts[teamname] = counts.get(teamname, 0) + 1
        if teamname in counts:
            counts[teamname] += 1
        else:
            counts[teamname]  = 0
            counts[teamname] += 1


    # Print each team's chances of winning, according to simulation
    for team in sorted(counts, key=lambda team: counts[team], reverse=True):
        print(f"{team}: {counts[team] * 100 / N:.1f}% chance of winning")


def simulate_game(team1, team2):
    """Simulate a game. Return True if team1 wins, False otherwise."""
    rating1 = team1["rating"]
    rating2 = team2["rating"]
    probability = 1 / (1 + 10 ** ((rating2 - rating1) / 600))
    return random.random() < probability


def simulate_round(teams):
    """Simulate a round. Return a list of winning teams."""
    winners = []

    # Simulate games for all pairs of teams
    for i in range(0, len(teams), 2):
        if simulate_game(teams[i], teams[i + 1]):
            winners.append(teams[i])
        else:
            winners.append(teams[i + 1])

    return winners


def simulate_tournament(teams):
    """Simulate a tournament. Return name of winning team."""
    #TODO
    tournateams = teams
    while True:
        winns = simulate_round(tournateams)
        if len(winns) == 1:
            return winns[0]
        else:
            tournateams = winns

if __name__ == "__main__":
    main()