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