我想这应该是一个常见的问题，但是我找不到解决方案：
让我们假设一个深度嵌套的列表，例如：
my_list <- list(
"first_node" = list(
"group_a" = list(
"E001" = 1:5,
"E002" = list(
"F001" = 6:10,
"F002" = 11:15
)
),
"group_b" = list(
"XY01" = list(
"Z1" = LETTERS[1:5],
"Z2" = LETTERS[6:10],
"Z3" = list(
"ZZ1" = LETTERS[1],
"ZZ2" = LETTERS[2],
"ZZ3" = LETTERS[3]
)
),
"YZ" = LETTERS[11:15]
),
"group_c" = list(
"QQQQ" = list(
"RRRR" = 200:300
)
)
),
"second_node" = list(
"group_d" = list(
"L1" = 99:101,
"L2" = 12
)
)
)
我想通过它们的名称检索元素，这些元素可能位于该列表的未知深度。重要的是，我只想要该特定元素，而是孩子，而不是父母。
例如，搜索my_list对于"XY01"应产生：
XY01 = list(
"Z1" = LETTERS[1:5],
"Z2" = LETTERS[6:10],
"Z3" = list(
"ZZ1" = LETTERS[1],
"ZZ2" = LETTERS[2],
"ZZ3" = LETTERS[3]
)
)
> str(XY01)
List of 3
$ Z1: chr [1:5] "A" "B" "C" "D" ...
$ Z2: chr [1:5] "F" "G" "H" "I" ...
$ Z3:List of 3
..$ ZZ1: chr "A"
..$ ZZ2: chr "B"
..$ ZZ3: chr "C"
最初，我想用来rapply()完成这项工作，但似乎无法访问names()当前迭代。我的第二次尝试是编写自定义递归函数：
recursive_extract <- function(haystack, needle){
lapply(names(haystack), function(x){
if (needle %in% names(haystack[[x]])) {
return(haystack[[needle]])
} else {
recursive_extract(haystack[[x]], needle)
}
}) %>% setNames(names(haystack))
}
...这似乎也是有问题的，因为lapply()即使NULL返回，它也总是会返回相同的对象，因此父级结构也随之而来。
我一直在寻找purrr和rlist-packages的便捷功能，但似乎大多数都不支持递归（？）。
提取所需的元素后，理想情况下，我希望选择要返回的子级别数。例如：desired_func(haystack, needle, get_depth = 1)对于前面的示例，将导致：
XY01 = list(
"Z1" = LETTERS[1:5],
"Z2" = LETTERS[6:10]
)
> str(XY01)
List of 2
$ Z1: chr [1:5] "A" "B" "C" "D" ...
$ Z2: chr [1:5] "F" "G" "H" "I" ...
非常感谢帮助！:)
这是一个函数，如果找到，它将返回第一个匹配项
find_name <- function(haystack, needle) {
if (hasName(haystack, needle)) {
haystack[[needle]]
} else if (is.list(haystack)) {
for (obj in haystack) {
ret <- Recall(obj, needle)
if (!is.null(ret)) return(ret)
}
} else {
NULL
}
}
find_name(my_list, "XY01")
我们避免lapply这样做，如果发现环路，可以尽早中断。
列表修剪实际上是一个单独的问题。最好用其他功能进行攻击。这应该工作
list_prune <- function(list, depth=1) {
if (!is.list(list)) return(list)
if (depth>1) {
lapply(list, list_prune, depth = depth-1)
} else {
Filter(function(x) !is.list(x), list)
}
}
那你可以做
list_prune(find_name(my_list, "XY01"), 1)
或用管道
find_name(my_list, "XY01") %>% list_prune(1)
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句