r - 如何使用 purrr 将计算列添加到嵌套数据框(列表列)

我想对嵌套数据框(存储为列表列)执行计算，并使用 purrr 函数将计算出的变量添加回每个数据框。我将使用此结果连接到其他数据，并保持紧凑有助于我更好地组织和检查它。我可以通过几个步骤来做到这一点，但似乎有一个我没有遇到过的解决方案。如果那里有解决方案，我无法轻松找到它。

加载库 。示例需要以下软件包(在 CRAN 上可用):

library(dplyr)
library(purrr)
library(RcppRoll) # to calculate rolling mean

示例数据 有 3 个对象，并且随着时间的推移重复测量:

test <- data_frame(
  id= rep(1:3, each=20),
  time = rep(1:20, 3),
  var1 = rnorm(60, mean=10, sd=3),
  var2 = rnorm(60, mean=95, sd=5)
  )

将数据存储为嵌套数据框 :

t_nest <- test %>% nest(-id)

     id              data
  <int>            <list>
1     1 <tibble [20 x 3]>
2     2 <tibble [20 x 3]>
3     3 <tibble [20 x 3]>

执行计算 。我将根据数据计算多个新变量，尽管稍后可以扩展一个解决方案。每次计算的结果将是一个数字矢量，其长度与输入的长度相同(n = 20):

t1 <- t_nest %>%
  mutate(var1_rollmean4 = map(data, ~RcppRoll::roll_mean(.$var1, n=4, align="right", fill=NA)),
         var2_delta4 = map(data, ~(.$var2 - lag(.$var2, 3))*0.095),
         var3 = map2(var1_rollmean4, var2_delta4, ~.x -.y))

     id              data var1_rollmean4 var2_delta4       var3
  <int>            <list>         <list>      <list>     <list>
1     1 <tibble [20 x 3]>     <dbl [20]>  <dbl [20]> <dbl [20]>
2     2 <tibble [20 x 3]>     <dbl [20]>  <dbl [20]> <dbl [20]>
3     3 <tibble [20 x 3]>     <dbl [20]>  <dbl [20]> <dbl [20]>

我的解决方案是先对此数据进行unnest，然后再次进行nest。这似乎没有任何问题，但似乎可能存在更好的解决方案。

t1 %>% unnest %>%
  nest(-id)

     id              data
  <int>            <list>
1     1 <tibble [20 x 6]>
2     2 <tibble [20 x 6]>
3     3 <tibble [20 x 6]>

另一个解决方案(from SO 42028710)很接近，但并非完全相同，因为它是一个列表，而不是嵌套的数据帧:

map_df(t_nest$data, ~ mutate(.x, var1calc = .$var1*100))

我使用 purrr Cheatsheet 找到了很多有用的信息，但找不到答案。

最佳答案

您可以在映射到数据列时包装另一个 mutate 并在每个嵌套的 tibble 中添加列:

t11 <- t_nest %>%
    mutate(data = map(data,
        ~ mutate(.x,
            var1_rollmean4 = RcppRoll::roll_mean(var1, n=4, align="right", fill=NA),
            var2_delta4 = (var2 - lag(var2, 3))*0.095,
            var3 = var1_rollmean4 - var2_delta4
        )
   ))

t11
# A tibble: 3 x 2
#     id              data
#  <int>            <list>
#1     1 <tibble [20 x 6]>
#2     2 <tibble [20 x 6]>
#3     3 <tibble [20 x 6]>

unnest-nest 方法，然后重新排列里面的列:

nest_unnest <- t1 %>%
    unnest %>% nest(-id) %>%
    mutate(data = map(data, ~ select(.x, time, var1, var2, var1_rollmean4, var2_delta4, var3)))

identical(nest_unnest, t11)
# [1] TRUE

关于r - 如何使用 purrr 将计算列添加到嵌套数据框(列表列)，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/46436107/