scikit learn dataset split code example

Example: scikit learn split data set

import numpy as np
from sklearn.model_selection import train_test_split
X, y = np.arange(10).reshape((5, 2)), range(5)
X





list(y)