Trianz interview question

Pandas