train_test_split - Creando librería de ML desde cero

Hola, les comparto una idea de como creo que podemos hacer nuestra propia función de train_test_split similar a sklearn, pero la idea es que nosotros aprendamos a crearla. Cada uno puede hacer sus implementaciones y mejoras. No pretendo crear un super código, sino simplemente algo sencillo e ilustrativo.