Рекурсивное преобразование фрейма данных во вложенный список, где уровень вложенности списка равен количеству столбцов в фрейме данных
У меня есть следующий фрейм данных.
Data_Frame <- structure(list(Factor_1 = c("AA", "AA", "AA", "AA", "AA", "AA", "AA", "AA", "AA", "AA", "AA", "AA", "AA", "AA", "AA", "BB", "BB", "BB", "BB", "BB", "BB", "BB", "BB", "BB", "BB", "BB", "BB", "BB", "BB", "BB", "CC", "CC", "CC", "CC", "CC", "CC", "CC", "CC", "CC", "CC", "CC", "CC", "CC", "CC", "CC", "DD", "DD", "DD", "DD", "DD", "DD", "DD", "DD", "DD", "DD", "DD", "DD", "DD", "DD", "DD"), Factor_2 = c("aa", "aa", "aa", "bb", "bb", "bb", "cc", "cc", "cc", "dd", "dd", "dd", "ee", "ee", "ee", "aa", "aa", "aa", "bb", "bb", "bb", "cc", "cc", "cc", "dd", "dd", "dd", "ee", "ee", "ee", "aa", "aa", "aa", "bb", "bb", "bb", "cc", "cc", "cc", "dd", "dd", "dd", "ee", "ee", "ee", "aa", "aa", "aa", "bb", "bb", "bb", "cc", "cc", "cc", "dd", "dd", "dd", "ee", "ee", "ee"), Factor_3 = c("xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz", "xxx", "yyy", "zzz")), class = "data.frame", row.names = c(NA, -60L))
Я хочу написать рекурсивную функцию, которая разобьет этот фрейм данных на вложенный список. Вывод должен выглядеть как следующий объект.
Split_Data <- lapply(lapply(split(Data_Frame, Data_Frame[, 1]), function (x) {
split(x, x[, 2])
}), function (x) {
lapply(x, function (y) {
split(y, y[, 3])
})
})
Другими словами, фрейм данных должен быть разделен сначала по значению первого столбца, затем по значению во втором столбце и т. д. и т. д., пока все столбцы не будут использованы для разделения данных на все меньшие и меньшие данные. фреймы (с каждым разделением список становится все более вложенным).
В этом примере во фрейме данных есть только три столбца, но на практике столбцов может быть любое количество, поэтому мне хотелось бы, чтобы рекурсивная функция могла обрабатывать любое количество столбцов.
base Предпочтительны решения R.
Спасибо!
Ответы 2
Я не совсем уверен, нужна ли вам помощь в написании рекурсивной функции или вы хотите иметь возможность динамически передавать количество столбцов для разделения.
Если последнее, вы можете просто сделать:
library(tidyverse)
col_names <- c('Factor_1', 'Factor_2', 'Factor_3')
Data_Frame %>%
group_by(across(all_of(!!col_names))) %>%
group_split()
В этом примере я вручную определяю три столбца, но в других случаях вы также можете автоматизировать это динамически, например:
col_names <- names(Data_Frame)[2:6]
или что-то в этом роде.
Вот простая рекурсивная функция для последовательного разделения фрейма данных по столбцам. Имейте в виду, что это будет работать плохо по мере увеличения количества столбцов, а также отдельных значений внутри столбцов.
recursive_split <- function(data, n = 1) {
if (n > ncol(data)) return(data)
lapply(split(data, data[[n]]), recursive_split, n + 1)
}
res <- recursive_split(Data_Frame)
identical(res, Split_Data)
[1] TRUE
Более производительная и гибкая рекурсивная функция разделения доступна в пакете collapse:
collapse::rsplit(Data_Frame, Data_Frame, keep.by = TRUE)
Похоже, это связано, учитывая использование
collapse::rsplitтам - stackoverflow.com/questions/76455797/…