calculate_average_friends.py

def get_friends_df(data: pd.DataFrame):
    """Get the number of friends for all people in the network"""

    all_people = list(set(data["person1"]).union(set(data["person2"])))
    num_friends = [
        {
            "person_id": person_id,
            "num_friends": get_num_friends(data, person_id),
            "avg_friends_of_friends": round(
                get_average_friends_of_a_person_friends(data, person_id), 2
            ),
        }
        for person_id in all_people
    ]
    return pd.DataFrame(num_friends)