group_split function

Split data frame by groups