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)
Sparkling Snake