R使用purrr对两次嵌套列表求和

Question

我有一个具有以下维度的 data.frame： 输出： as_tibble(data2)

lamda meanlog sdlog freq      freqsev   
<dbl>   <dbl> <dbl> <list>    <list>    
1     5       9   2   <int [4]> <list [4]>
2     2      10   2.1 <int [4]> <list [4]>
3     3      11   2.2 <int [4]> <list [4]>

其中freqsev 是长度为freq 的值的列表，freq 本身是长度为s 的值的列表，其中s 是模拟次数。

library(tidyverse)
set.seed(123)
s <- 5
data <- data.frame(lamda = c(5, 2, 3), meanlog = c(9, 10, 11), sdlog = c(2, 2.1, 2.2))

data2 <- data %>% mutate(
  freq = map(lamda, ~rpois(s, .x)),
  freqsev = map(freq, ~map(.x, function(k) rlnorm(k, meanlog, sdlog)))
  )

我想对freqsev 求和（产生<dbl [4]>，其中[4] 是s 的索引），即freq 出现次数的总和，例如

对于data2$freqsev[[1]][[1]]，我希望得到总和。

如何做到这一点？谢谢。

Answer 1

说实话，这是一种非常复杂的数据存储方式，在创建 freq 列之后使用 unnest() 可能会更好。但是，您可以像这样得到 freqsev 向量的总和：

data2 <- data %>% mutate(
  freq = map(lamda, ~rpois(s, .x)),
  freqsev = map(freq, ~map(.x, function(k) rlnorm(k, meanlog, sdlog))),
  freqsum = map(freqsev, ~map_dbl(.x, ~sum(.x)))
)

因为freqsev是一个双嵌套列表，所以还需要双映射求和操作。