Data Science Code Example Python