Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import csv
- def display_manager_solution(csv_data):
- employees = []
- managers = {}
- current_id = 1
- reader = csv.DictReader(csv_data.splitlines())
- for row in reader:
- manager_full_name = row['Manager']
- if manager_full_name and manager_full_name not in managers:
- managers[manager_full_name] = str(current_id)
- current_id += 1
- # Assign an ID to the employee (who is not already a manager)
- employee_full_name = f"{row['FirstName']} {row['LastName']}"
- if employee_full_name not in managers:
- managers[employee_full_name] = str(current_id)
- current_id += 1
- manager_id = managers.get(manager_full_name, '')
- employees.append((managers[employee_full_name], manager_id, row['LastName'], row['FirstName']))
- employees.sort(key=lambda x: x[0])
- for employee in employees:
- print(','.join(employee))
- def main():
- csv_data = []
- with open('employees.csv') as f:
- csvreader = csv.reader(f)
- csv_data = list(csvreader)
- # Display the manager solution
- display_manager_solution(csv_data)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement