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