Lyft interview question

Write an algorithm for stratified split of data