EDA in python