結果 : python split data into training and test set