Slalom interview question

Python Take Away Assignment related to ML