python coding practice for data science interview