# ------------------------------------------------------------------- # Modules # ------------------------------------------------------------------- import random # ------------------------------------------------------------------- # Constants # ------------------------------------------------------------------- COUNTRIES = ["Great Britain", "United States", "China", "Norway"] FILENAME = "medals.csv" # ------------------------------------------------------------------- # Global variables # ------------------------------------------------------------------- number_of_countries = 0 gold_medals = [] silver_medals = [] bronze_medals = [] file = None # file to write to i = 0 # ------------------------------------------------------------------- # Subprograms # ------------------------------------------------------------------- def generate_medal_data(pNumberOfCountries): for i in range(pNumberOfCountries): gold = random.randint(0, 50) gold_medals.append(gold) silver = random.randint(0, 50) silver_medals.append(silver) bronze = random.randint(0, 50) bronze_medals.append(bronze) # ------------------------------------------------------------------- # Main program # ------------------------------------------------------------------- number_of_countries = len(COUNTRIES) generate_medal_data(number_of_countries) print("Writing medal data to " + FILENAME) file = open(FILENAME, "w") file.write("Country,Gold,Silver,Bronze\n") # iterate through every country for i in range(number_of_countries): file.write(COUNTRIES[i] + ",") file.write(str(gold_medals[i]) + ",") file.write(str(silver_medals[i]) + ",") file.write(str(bronze_medals[i]) + "\n") file.close() # Challenges: # 1) Calculate the total number of medals for each country # 2) Write the total number of medals for each country to the CSV file # 3) Display the overall number of medals to the screen