結果 : how to split numpy array into train and test