Выполняя программу "Обмен студентами", университету необходимо создать n групп студентов. В каждой группе долж

from enum import Enum
import itertools
 
class Speciality(Enum):
    RPIS = 1
    PO = 2
    UITC = 3
 
class Student:
    def __init__(self, name, speciality):
        self.name = name
        self.speciality = speciality
 
students = [
    Student("Иванов", Speciality.RPIS),
    Student("Кузнецов", Speciality.PO),
    Student("Петров", Speciality.PO),
    Student("Сидоров", Speciality.UITC),
    Student("Смирнов", Speciality.RPIS)
]
 
students.sort(key=lambda x: x.name)  # сортировка по фамилиям
 
groups = [[] for _ in range(2)]  # создание двух групп
 
# создание словаря со списками студентов каждой специальности
students_by_speciality = {spec: [] for spec in Speciality}
for student in students:
    students_by_speciality[student.speciality].append(student)
 
# равномерное распределение студентов по группам
for spec_students in students_by_speciality.values():
    for student, group in zip(spec_students, itertools.cycle(groups)):
        group.append(student)
 
# вывод списка групп со студентами
for i, group in enumerate(groups):
    print(f"Группа {i + 1}: ", end="")
    for student in group:
        print(f"{student.name} ({student.speciality.name}), ", end="")
    print()