Python to perform data analysis