R сгладить список иерархии в матрицу или data.frame

Я хотел бы сгладить иерархию списка (похожую на JSON) на матрицу или фрейм данных. Скажем, что я создаю следующий список:

a <- list(
  b1 = list(
     c1 = list(
       d1 = data.frame()
     ),
     c2 = data.frame()
  ),
  b2 = data.frame()
)

Где каждая буква является другим уровнем или понижает иерархию. Затем мне нужна функция, например. listToMatrix(mylist = a, steps = 2), который генерирует следующее:

    [,1] [,2]
[1,] "b1" "c1"
[2,] "b1" "c2"
[3,] "b2" "b2"

Обратите внимание, что аргумент функции steps = 2 означает, что он должен идти только на 2 ступени вниз по иерархии. Кроме того, если в одном направлении недостаточно уровней, см. b2, тогда оно должно содержать предыдущее имя списка в матрице.

Любые предложения?:)

Ответы

Ответ 1

Вот решение. Поэтому он легко читается здесь, я разбил код на две части. Позже вы можете легко объединить две части в одну функцию.

Во-первых, функция, которая получает матрицу всех имен, используя рекурсию:

anames <- function(x) {

   require(plyr)
   if (is.data.frame(x)) return(NA)

   y <- do.call(rbind.fill.matrix,
                mapply(cbind, names(x), lapply(x, anames),
                       SIMPLIFY = FALSE))
   colnames(y) <- NULL

   return(y)
}

anames(a)
#      [,1] [,2] [,3] [,4]
# [1,] "b1" "c1" "d1" NA  
# [2,] "b1" "c2" NA   NA  
# [3,] "b2" NA   NA   NA

Затем, функция, которая применяет данный ввод steps, и заполняет NA, как вы просили:

listToMatrix <- function(myList, steps = Inf) {

   a <- anames(myList)
   steps <- min(steps, ncol(a) - 1)
   cols.idx <- seq_len(steps)
   a <- a[, cols.idx]
   for (j in tail(cols.idx, -1))
      a[, j] <- ifelse(is.na(a[, j]), a[, j - 1], a[, j])

   return(a)
}

listToMatrix(a, 2)
#      [,1] [,2]
# [1,] "b1" "c1"
# [2,] "b1" "c2"
# [3,] "b2" "b2"