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