R/compat-tidyr.R

In krlmlr/styler: Non-Invasive Pretty Printing of R Code

nest_ <- function(data, key_col, nest_cols = character()) {
  key_column <- setdiff(names(data), nest_cols)
  key_data <- data[[key_column]]
  key_levels <- unique(key_data)
  key_factor <- factor(key_data, levels = key_levels)

  res <- vec_split(data[, nest_cols], key_factor)
  names(res) <- c(key_column, key_col)
  res
}