如何使用python (google colab)将人员分配到房间？

people = {
    "person1": {"name": "John", "age": 25},
    "person2": {"name": "Jane", "age": 30},
    ...
}

rooms = {
    "room1": {"name": "Living Room", "capacity": 5},
    "room2": {"name": "Bedroom", "capacity": 2},
    ...
}

def assign_rooms(people, rooms):
    assignments = {}
    for person_id, person_info in people.items():
        assigned = False
        for room_id, room_info in rooms.items():
            if room_info["capacity"] > 0:
                assignments[person_id] = room_id
                room_info["capacity"] -= 1
                assigned = True
                break
        if not assigned:
            assignments[person_id] = "Unassigned"
    return assignments

# 测试数据
people = {
    "person1": {"name": "John", "age": 25},
    "person2": {"name": "Jane", "age": 30},
    "person3": {"name": "Mike", "age": 35},
}

rooms = {
    "room1": {"name": "Living Room", "capacity": 2},
    "room2": {"name": "Bedroom", "capacity": 1},
}

# 分配房间
assignments = assign_rooms(people, rooms)

# 打印结果
for person_id, room_id in assignments.items():
    print(f"{people[person_id]['name']} is assigned to {rooms[room_id]['name']}")