I am doing a feature extraction for ML project. I have to split my data features and labels for training model. For this I wrote a code as shown. After grouped columns with