結果 : create train and test data in python