r - 将函数应用于R中嵌套列表的某些元素

Question

我希望仅将函数应用于嵌套列表的某些元素

l <- list()

l$a$forecast <- rnorm(3)
l$a$model <- "arima"
l$b$forecast <- rnorm(3)
l$b$model <- "prophet"

所需的输出将是这样的：将 sum 函数应用于列表的 $forecast 元素

fcst <- c(sum(l$a$forecast),sum(l$b$forecast))
mdl <- c(l$a$model,l$b$model)
df <- data.frame(fcst,mdl)

我试过这样的事情：

df <- lapply(l$forecast, function(x) sum(x))
df <- do.call(rbind, Map(cbind, sku = names(df)))

Answer 1

---

do.call(
  rbind,
  lapply(
    l,
    function(x) list(fcst = sum(x$forecast), model = x$model)
  )
)

由于您知道返回对象的确切尺寸，因此您可以vapply在这种情况下使用，以实现较小的性能改进：

vapply(
  l,
  FUN = function(x) list(fcst = sum(x$forecast), model = x$model),
  FUN.VALUE = list(fcst = numeric(1), model = character(1))
)

但是，生成的对象可能很难使用。

Answer 2

rrapply()在-package 中使用的另一种方法rrapply与 dplyr's 结合使用bind_rows()。这也扩展到包含更深嵌套级别的列表。

rrapply::rrapply(l, condition = function(x, .xname) .xname == "forecast", f = sum) %>%
  dplyr::bind_rows()

#> # A tibble: 2 x 2
#>   forecast model  
#>      <dbl> <chr>  
#> 1    -1.28 arima  
#> 2     1.10 prophet

数据

set.seed(1)

l <- list(
  a = list(forecast = rnorm(3), model = "arima"),
  b = list(forecast = rnorm(3), model = "prophet")
)

Answer 3

您可以使用 object 获取字母letters，然后在循环中使用其输出：

    n = 2 #number of lists you have
    sumfore = model = vector()
    for(i in letters[seq(1,n,1)]){
      sumfore[i] = sum(l[[i]]$forecast)
      model[i] =l[[i]]$model}
    newdf = data.frame(sumfore, model)