結果 : divide data into train and test in python