从嵌套列表中按名称提取元素

Question

对于一个有名称的嵌套列表，最好的提取特定元素的方法是什么？如果我有一个已知字段的列表（例如来自yaml文件），我想提取一个元素（列表或其他）而不必搜索名称和索引或尝试跟踪str输出中的层级。

例如，我知道lm返回一个包含qr信息的嵌套列表。

fit <- lm(mpg ~ wt, mtcars)
fit$qr$qraux
# [1] 1.176777 1.046354

但如果我不知道顺序，我只想指定列表以及元素的名称。理想情况下，某些东西会给我元素的索引路径和名称路径以及元素本身。

相关问题：链接，链接，链接。

- rawr

---

看看purrr包。它就是为此而设计的。对于线性模型，你还可以看看broom包。你会喜欢它的。 - TheRimalaya

另一个相关的问题：https://dev59.com/_Ybca4cB1Zd3GeqPac21 - Karolis Koncevičius

Answer 1

我的递归版本1开始变得比我最初想象的更加有错误，所以我采取了一个简单的方法，基本上是在utils:::print.ls_str的捕获输出中搜索筛选（我认为是这样）。

到目前为止，这种方法至少有两个缺点：输出的捕获和eval-parse-texting，但对于一个非常嵌套的列表，例如ggplot2::ggplotGrob，它似乎可以正确地工作。

这些只是一些辅助函数。

unname2 <- function(l) {
  ## unname all lists
  ## str(unname2(lm(mpg ~ wt, mtcars)))
  l <- unname(l)
  if (inherits(l, 'list'))
    for (ii in seq_along(l))
      l[[ii]] <- Recall(l[[ii]])
  l
}

lnames <- function(l) {
  ## extract all list names
  ## lnames(lm(mpg ~ wt, mtcars))
  nn <- lpath(l, TRUE)
  gsub('\\[.*', '', sapply(strsplit(nn, '\\$'), tail, 1))
}

lpath <- function(l, use.names = TRUE) {
  ## return all list elements with path as character string
  ## l <- lm(mpg ~ wt, mtcars); lpath(l); lpath(l, FALSE)
  ln <- deparse(substitute(l))
  # class(l) <- NULL
  l <- rapply(l, unclass, how = 'list')
  L <- capture.output(if (use.names) l else unname2(l))
  L <- L[grep('^\\$|^[[]{2,}', L)]
  paste0(ln, L)
}

这个返回有用信息的函数

lextract <- function(l, what, path.only = FALSE) {
  # stopifnot(what %in% lnames(l))
  ln1 <- eval(substitute(lpath(.l, TRUE), list(.l = substitute(l))))
  ln2 <- eval(substitute(lpath(.l, FALSE), list(.l = substitute(l))))
  cat(ln1[idx <- grep(what, ln1)], sep = '\n')
  cat('\n')
  cat(ln2[idx], sep = '\n')
  cat('\n')
  if (!path.only)
    setNames(lapply(idx, function(x) eval(parse(text = ln1[x]))), ln1[idx])
  else invisible()
}

fit <- lm(mpg ~ wt, mtcars)
lextract(fit, 'qraux')
# fit$qr$qraux
# 
# fit[[7]][[2]]
# 
# [1] 1.176777 1.046354

因此，我可以直接使用返回值，或者现在我有了索引。

fit[[7]][[2]]
# [1] 1.176777 1.046354


## etc
lextract(fit, 'qr', TRUE)

# fit$qr
# fit$qr$qr
# fit$qr$qraux
# fit$qr$pivot
# fit$qr$tol
# fit$qr$rank
# 
# fit[[7]]
# fit[[7]][[1]]
# fit[[7]][[2]]
# fit[[7]][[3]]
# fit[[7]][[4]]
# fit[[7]][[5]]

然而，我更倾向于内置或一行代码的解决方案。

- rawr

Answer 2

这里是另一种递归尝试。我不确定输出应该如何结构化，但我认为此方法提供了足够的信息以提取其余部分。

此处返回的值是一个索引向量和元素长度。因此，对于fit示例，它返回c(inds=7, len=5)，对应于fit中的第7个位置，元素长度为5。

rnames <- function(lst, item) {
  f <- function(ll, inds) {
    if ((ii <- match(item, names(ll), FALSE)))
      list(inds=c(inds, ii), len=length(ll[[ii]]))
    else if (all(is.atomic(unlist(ll, FALSE))) || !is.list(ll))
      NULL
    else
      lapply(seq_along(ll), function(i) f(ll[[i]], inds=c(inds, i)))
  }
  unlist(f(lst, NULL))
}

rnames(fit, "qr")
# inds  len 
#    7    5 

这只会找到第一个匹配项，因此，如果列表中有多个具有相同名称的元素，则它将返回第一个匹配项的索引。下面是稍微嵌套一些的示例，只有第一个“d”会被返回。

lst <- list(
  "a"=list("b"=1, "c"=2, "d"=list(1:5)), 
  "d"=list("f"=5),
  "g"=list("h"=list("i"=1:5), "k"=list(1:3, list(letters[1:4])))
)

rnames(lst, "d")
# inds  len 
#    2    1 

当嵌套层数较多时

rnames(lst, "k")
# inds1 inds2   len 
#     3     2     2 

## So, that would correspond to 
lst[[3]][[2]][1:2]

- Rorschach