結果 : splitting training and test data in python