結果 : divide dataset into train and test python