Data Analysis Python Code