結果 : split data to train and test python