Question

How to add class objects to a dictionary? *PYTHON ONLY* NO IMPORT RANDOM

Write a class named Employee that has data members for an employee's name, ID_number, salary, and email_address (you must use those names - don't make them private). Write a function named make_employee_dict (outside of the Employee class) that takes as parameters a list of names, a list of ID numbers, a list of salaries and a list of email addresses. The function should take the first value of each list and use them to create an Employee object. It should do the same with the second value of each list, etc. to the end of the lists. As it creates these objects, it should add them to a dictionary, where the key is the ID number and the value for that key is the whole Employee object. The function should return the resulting dictionary.

This is what I got so far:

class Employee:

    """
    Represents the employee's name, ID_number, salary, and email_address.
    """
    def __init__(self, name, ID_number, salary, email_address):
        self.name = name
        self.ID_number = ID_number
        self.salary = salary
        self.email_address = email_address

def make_employee_dict(names, IDs, salaries, emails):
    """
    This function "make_employee_dict" takes the first value of each list and uses them to create
    an Employee object. It does the same with the second value of each list, etc. to the end of the
    the lists. This where I'm stuck at :/
    """

If I input:

names = ["Ann", "Gabriel", "Maria"]
IDs = [10, 11, 12]
salaries = [35, 29, 35]
emails = ["ann@blah.com", "gabriel@blah.com", "maria@blah.com"]
print(make_employee_dict(names, IDs, salaries, emails))

Expected output:

{10: 'Ann', 10, 35, 'ann@blah.com', 11: 'Gabriel', 11, 29, 'gabriel@blah.com', 12: 'Maria', 35, 'maria@blah.com'}

Answer 1

# do comment if any problem arises

# Code

class Employee:

    """

    Represents the employee's name, ID_number, salary, and email_address.

    """

    def __init__(self, name, ID_number, salary, email_address):

        self.name = name

        self.ID_number = ID_number

        self.salary = salary

        self.email_address = email_address

    def __repr__(self):

        return str(self.name)+", " + str(self.ID_number)+", "+str(self.salary)+", "+str(self.email_address)

def make_employee_dict(names, IDs, salaries, emails):

    """

    This function "make_employee_dict" takes the first value of each list and uses them to create

    an Employee object. It does the same with the second value of each list, etc. to the end of the

    the lists. This where I'm stuck at :/

    """

    # create empty dictionary

    employee_dict = {}

    for i in range(len(names)):

        # create an employee

        employee = Employee(names[i], IDs[i], salaries[i], emails[i])

        # add to employee dictionary

        employee_dict[IDs[i]] = employee

    return employee_dict

# input

names = ["Ann", "Gabriel", "Maria"]

IDs = [10, 11, 12]

salaries = [35, 29, 35]

emails = ["ann@blah.com", "gabriel@blah.com", "maria@blah.com"]

print(make_employee_dict(names, IDs, salaries, emails))

Screenshot:

Output: