結果 : divide train and test data in python