結果 : python split data in train and test