Customer Analysis — Interview Query pandas problem