如何轻松迭代嵌套列表的叶子

时间:2018-04-17 14:50:10

标签: r

假设我有以下列表:

a <- list(a = 1, b = list(a = list(b = 1, c = 2), c = 1), c = 1)

我想删除所有名称为c的叶子。我该如何以惯用的方式做到这一点?这似乎是一个适合rapply的工作但是使用rapply我无法找到一种方法来访问我正在迭代的列表元素的名称。

2 个答案:

答案 0 :(得分:1)

这个怎么样?

a <- list(a = 1, b = list(a = list(b = 1, c = 2), c = 1), c = 1)

aunls <- unlist(a)
aunls[!grepl('c', names(aunls))]

然后你可以再次使用名称结构进行嵌套或使用lapply。

答案 1 :(得分:1)

您可以使用递归lapply函数执行此操作,例如:

foo <- function(x, name) {
  if(any(idx <- names(x) == name)) x <- x[!idx]
  if(is.list(x)) lapply(x, foo, name) else x
}

foo(a, "c")
# $a
# [1] 1
# 
# $b
# $b$a
# $b$a$b
# [1] 1