結果 : split data in training and test python