Friend Group data structure and accompanying functions from COMP0233 session 14/10/2024

group.py

@@ -1,5 +1,133 @@
 """An example of how to represent a group of acquaintances in Python."""
 
 # Your code to go here...
+import numpy as np
+import json
+import yaml
+"""
+rows = [{ "name": "", "age": "", "job": "", Relationships: {} }, 
+        ]
+"""
+rows = [
+        { "name": "Jill", "age": 26, "job": "biologist", "Relationships": {"Zalika": "Friend", "John": "Partner"}},
+
+        { "name": "Zalika", "age": 28, "job": "artist", "Relationships": {"Jill": "Friend", "Nash": "Landlord"}}, 
+
+        {"name": "John", "age": 27, "job": "writer", "Relationships": {"Jill": "Partner", "Nash": "Cousin"}}, 
+
+        { "name": "Nash", "age": 34, "job": "chef", "Relationships": {"John": "Cousin", "Zalika": "Tenant"}},          
+        ]
+
+def add_person(name, age, job=None, relationships=None):
+    """
+    Adds a new person to the group.
+    Note that relationships will be a dictionary, with the other person being the key and then the relationship, e.g. {"Person's Name": "type of relationship"}
+    Example:
+    add_person("Dave", 63, relationships={"Jill": "Dad", "John": "Father in law"})
+    """
+    person = {"name": name,
+        "age": age,
+        "job": job if job else None,
+        "relationships" : relationships if relationships  else {}
+        }
+    my_group.append(person)
+    print(f"{name} added to the group")
+
+def get_average_ages(my_group):
+    """
+    Calculates the average (mean) age of the group
+    """
+    ages = [person["age"] for person in my_group]
+    return np.mean(ages)
+
+def forget_person(person1, person2):
+    """
+    Removes a relationship between two people, ensuring it is removed from both sides.
+    """
+    for person in my_group:
+        if person['name'] == person1 and person2 in person["Relationships"]:
+            del person["Relationships"][person2]
+    for person in my_group:
+        if person['name'] == person2 and person1 in person["Relationships"]:
+            del person["Relationships"][person1]
+
+def safe_get(person, key):
+    """
+    Safely retrieves the value for the specified key, returning None if the key does not exist.
+    """
+    try:
+        return person[key]
+    except KeyError:
+        return None
+
+# Function to save the group to a JSON file
+def save_to_json(filename, data):
+    with open(filename, 'w') as file:
+        json.dump(data, file, indent=4)
+    print(f"Data saved to {filename}")
+
+# Function to load the group from a JSON file
+def load_from_json(filename):
+    with open(filename, 'r') as file:
+        data = json.load(file)
+    return data
+
+# Function to save the group to a YAML file
+def save_to_yaml(filename, data):
+    with open(filename, 'w') as file:
+        yaml.dump(data, file, default_flow_style=False)
+    print(f"Data saved to {filename}")
+
+# Function to load the group from a YAML file
+def load_from_yaml(filename):
+    with open(filename, 'r') as file:
+        data = yaml.safe_load(file)
+    return data
+
+
+if __name__ == "__main__":
+    # Initialize the group
+    my_group = rows.copy()
+
+    # Calculate the average age of the group
+    mean_ages = get_average_ages(my_group)
+    print("Average age of group = ", mean_ages)
+
+    # Calculate and print the maximum age in the group
+    max_age = max(person["age"] for person in my_group)
+    print('Maximum age = ', max_age)
+
+    # Calculate the average number of relationships per person
+    num_relationships = [len(person["Relationships"]) for person in my_group]
+    print("Average number of relationships = ", np.mean(num_relationships))
+
+    # Find the maximum age of people with at least one relationship
+    max_age_with_relations = max(person["age"] for person in my_group if len(person["Relationships"]) >= 1)
+    print('Maximum age of people with at least one relationship = ', max_age_with_relations)
+
+    # Advanced: Find the maximum age of people who have at least one friend
+    max_age_with_friends = max(
+        person["age"] for person in my_group if sum(1 for value in person["Relationships"].values() if value == "Friend") >= 1
+    )
+    print('Maximum age of people with at least one friend = ', max_age_with_friends)
+
+    # Example usage
+    filename_json = 'group_data.json'
+    save_to_json(filename_json, my_group)
+
+    # Load data from the JSON file
+    loaded_group_json = load_from_json(filename_json)
+
+    # Check if the original data and the loaded data are identical
+    print("Data is identical:", my_group == loaded_group_json)
+
+    # Example usage
+    filename_yaml = 'group_data.yaml'
+    save_to_yaml(filename_yaml, my_group)
+
+    # Load data from the YAML file
+    loaded_group_yaml = load_from_yaml(filename_yaml)
+
+    # Check if the original data and the loaded data are identical
+    print("Data is identical:", my_group == loaded_group_yaml)
 
-my_group =