Machine Learning data preprocessing (one hot encoding for categorical variables using python)