結果 : python code to split data into train and test