จงหาจำนวนแผนกต่างๆ ในแต่ละโรงเรียนที่มีจำนวนแผนกต่างๆ น้อยกว่า 5 แผนก
Find the number of different departments in each school whose number of different departments is less than 5.
df_dept = pd.DataFrame(columns = ['school_code', 'dept_name'])
df_dept.groupby('school_code')['dept_name'].nunique().loc[lambda x : x<5]