10. Find the employees with their primary department | Pyspark | SQL Solution

[@sabesanj5509]

SELECT employee_id, department_id
FROM employees
GROUP BY employee_id
HAVING COUNT(employee_id) = 1
UNION
SELECT employee_id, department_id
FROM employees
WHERE primary_flag = 'Y';