Arrow Thought interview question

python with data science related questions