generated from mwc/lab_weather
124 lines
3.3 KiB
Python
124 lines
3.3 KiB
Python
# friend_functions.py
|
|
# ------------
|
|
# By MWC Contributors
|
|
#
|
|
# Each of the functions below expects a list of dictionaries as its first
|
|
# argument. Two examples of the expected input are provided in people.family
|
|
# and people.friends.
|
|
|
|
# Your job is to complete these functions. Remove the NotImplementedError from
|
|
# each and instead write code which returns the expected values.
|
|
|
|
def count_people(people):
|
|
"""Counts the number of people.
|
|
|
|
>>> count_people(family)
|
|
5
|
|
>>> count_people(friends)
|
|
10
|
|
"""
|
|
n = 0
|
|
for i in people:
|
|
n = n + 1
|
|
return n
|
|
|
|
def get_email(people, name):
|
|
"""Returns the named person's email address. If there is no such person, returns None.
|
|
|
|
>>> get_email(family, "Tad Winters")
|
|
"ligula.aenean@hotmail.edu"
|
|
>>> get_email(friends, "Tad Winters")
|
|
None
|
|
"""
|
|
for person in people:
|
|
if person["name"] == name:
|
|
return person["email"]
|
|
return None
|
|
|
|
def count_favorite_colors(people, name):
|
|
"""Returns the number of colors liked by the named person. If there is no such person, returns None.
|
|
|
|
>>> count_favorite_colors(family, "Tad Winters")
|
|
2
|
|
>>> count_favorite_colors(family, "Raphael Chambers")
|
|
1
|
|
"""
|
|
for person in people:
|
|
if person["name"] == name:
|
|
return len(person["favorite_colors"])
|
|
return None # If the person is not found
|
|
|
|
def people_who_like_color(people, color):
|
|
"""Returns a list containing only those people who like the given color.
|
|
>>> people_who_like_color(family, "yellow")
|
|
[
|
|
{
|
|
"name": "Walker Hurley",
|
|
"email": "auctor.odio@icloud.ca",
|
|
"favorite_colors": ["red", "yellow", "blue", "orange"],
|
|
},
|
|
{
|
|
"name": "Clementine Joseph",
|
|
"email": "hendrerit@aol.co.uk",
|
|
"favorite_colors": ["yellow", "aqua", "black"],
|
|
}
|
|
]
|
|
>>> people_who_like_color(family, "indigo")
|
|
[]
|
|
"""
|
|
matching_people = []
|
|
for person in people:
|
|
if color in person["favorite_colors"]:
|
|
matching_people.append(person)
|
|
return matching_people
|
|
|
|
def count_people_who_like_color(people, color):
|
|
"""Returns the number of people who like a given color.
|
|
|
|
>>> count_people_who_like_color(family, "red")
|
|
2
|
|
>>> count_people_who_like_color(family, "orange")
|
|
1
|
|
"""
|
|
n = 0
|
|
for person in people:
|
|
for colors in person["favorite_colors"]:
|
|
if colors == color:
|
|
n = n + 1
|
|
return n
|
|
|
|
|
|
def get_color_dict(people):
|
|
"""Returns a dict showing how many people like each color.
|
|
Any color liked by any of the people will be included, and only colors
|
|
liked by someone will be included. The order of items in the dict doesn't matter.
|
|
|
|
>>> get_color_dict(family)
|
|
{
|
|
"aqua": 2,
|
|
"red": 2,
|
|
"blue": 2,
|
|
"black": 2,
|
|
"white": 1,
|
|
"grey": 1,
|
|
"yellow": 2,
|
|
"orange": 1,
|
|
}
|
|
"""
|
|
color_dict = {}
|
|
|
|
for person in people:
|
|
for color in person["favorite_colors"]:
|
|
if color in color_dict:
|
|
color_dict[color] = color_dict[color] + 1
|
|
else:
|
|
color_dict[color] = 1
|
|
return color_dict
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|