!/usr/bin/python

import numpy as np
from sklearn.model_selection import train_test_split

X, y = np.arange(10).reshape((5, 2)), range(5)

print("X:{} \n", X)
print("y: {} \n", list(y))

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=42)

print("X_train:{}\n", X_train)
print("X_test:{}\n", X_test)
print("y_train:{} \n", y_train)
print("y_test:{}\n", y_test)

print(train_test_split(y, shuffle=False))