Split numpy array into train and test. So before scaling, I need to split the data.