diff options
Diffstat (limited to 'wk7/lect/favorites.py')
| -rw-r--r-- | wk7/lect/favorites.py | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/wk7/lect/favorites.py b/wk7/lect/favorites.py index 289a17f..446a92e 100644 --- a/wk7/lect/favorites.py +++ b/wk7/lect/favorites.py @@ -7,15 +7,15 @@ import csv with open("favorites.csv") as file: reader = csv.DictReader(file) - scratch, c, python = 0, 0, 0 + counts = {} for row in reader: - favorite = row['language'] - if favorite == "Scratch": - scratch += 1 - elif favorite == "C": - c += 1 - elif favorite == "Python": - python += 1 - -print("Scratch:", scratch) + favorite = row["language"] + if favorite in counts: + counts[favorite] += 1 + else: + counts[favorite] = 1 + +for favorite in counts: + print(favorite, ':', counts[favorite]) + |