import pandas as pd

df = pd.read_csv('data/enrolment_3.csv')

# room number column 생성 및 not assigned 일괄 적용 & 배정된 강의실 정보 LIST 생성 
df['room number'] = 'not assigned'
rooms = ['Auditorium', 'Large room', 'Medium room', 'Small room']

for room in rooms:
    # 강의실 종류별로 배정된 과목들을 찾아, 오름차순 정리 
    courses = sorted(list((df.loc[df['room assignment'] == room, 'course name']).value_counts().index))
    for course in range(len(courses)):
        # 강의실명, 과목명 조건에 맞는 row 를 찾기 위한 filter 생성
        condition = (df['room assignment'] == room) & (df['course name'] == courses[course])
        # 필터 조건에 따라 room number 의 값을 변경 
        # 강의실 리스트상  'not assigned' 항목은 포함되지 않아, 해당 항목들은 변경되지 않고 'not assigned'로 유지됨
        df.loc[condition, 'room number'] = f'{(room.split(" "))[0]}-{course + 1}'
    
# 기존 'room assignment' column 삭제 
df.drop('room assignment', axis='columns', inplace=True)

df