Pregunta de entrevista de Lyft

Write an algorithm for stratified split of data