Arrk Interview Question

Pandas implementation, lambda function