ค้นหารหัสของแผนกที่ผู้จัดการคนใดคนหนึ่งจัดการพนักงาน 4 คนขึ้นไป
Find the ids of the departments where any manager is managing 4 or more employees.
df = pd.DataFrame(columns=['department_id', 'manager_id', 'employee_id'])
df.groupby(['department_id', 'manager_id']).size()[lambda x: x >= 4].reset_index()['department_id']