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