結果 : how to create train and test data in python