R 在名称后面的列表中添加一个层

发布于11月09日

我有这个单层列表:

have <- list("a" = 1, "b.a" = 2, "b.b" = 3)

我需要这两层列表:

need <- list("a" = 1, "b" = list("a" = 2, "b" = 3))

如果解决方案可以是递归的就更好了.我们的目标是将数据 struct 化，因为它可以被解析为一些JSON文件.

谢谢！

推荐答案

这个递归函数应该可以做到这一点:

f <- function(obj) {
  nms <- lapply(strsplit(names(obj), '\\.'), function(x){
    if(length(x) == 1) x else c(x[1], paste(x[-1], collapse = '.'))
  })
  first_nms <- sapply(nms, `[`, 1)
  last_nms <- sapply(nms, `[`, -1)
  un_nms <- unique(first_nms)
  setNames(lapply(un_nms, function(x) {
    o <- unname(obj[first_nms %in% x])
    if(length(o) == 1) unlist(o) else
      f(setNames(o, last_nms[first_nms %in% x]))
    }), un_nms)
}

测试，我们有:

have <- list("a" = 1, "b.a" = 2, "b.b" = 3)

f(have)
#> $a
#> [1] 1
#> 
#> $b
#> $b$a
#> [1] 2
#> 
#> $b$b
#> [1] 3

我们可以看到它在深度嵌套的列表上工作:

f(list(a = 1, b.a = 2, b.b = 3, c.a.a = 4, c.a.b = 5, c.b.a = 6, c.b.b = 7))
#> $a
#> [1] 1
#> 
#> $b
#> $b$a
#> [1] 2
#> 
#> $b$b
#> [1] 3
#> 
#> 
#> $c
#> $c$a
#> $c$a$a
#> [1] 4
#> 
#> $c$a$b
#> [1] 5
#> 
#> 
#> $c$b
#> $c$b$a
#> [1] 6
#> 
#> $c$b$b
#> [1] 7