Train Test Split in Cross Validation using Pandas and Numpy