Created
May 26, 2023 11:44
-
-
Save andrija-zikovic/66f3928ef64ce8b369dd134bf464393f to your computer and use it in GitHub Desktop.
CS50/world-cup
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# Simulate a sports tournament | |
import csv | |
import sys | |
import random | |
# 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 | |
with open(sys.argv[1], 'r') as file: | |
reader = csv.DictReader(file) | |
for row in reader: | |
row["rating"] = int(row["rating"]) | |
teams.append(row) | |
counts = {} | |
# TODO: Simulate N tournaments and keep track of win counts | |
for i in range(N): | |
winner = simulate_tournament(teams) | |
if winner in counts: | |
counts[winner] += 1 | |
else: | |
counts[winner] = 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 | |
while len(teams) > 1: | |
teams = simulate_round(teams) | |
return teams[0]["team"] | |
if __name__ == "__main__": | |
main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment