如何使用 purrr 旋转嵌套数据框？答案

【问题标题】：How can I use purrr to pivot a nested dataframe?如何使用 purrr 旋转嵌套数据框？
【发布时间】：2021-03-26 18:29:28
【问题描述】：

下面的代码创建了数据框的简化版本，并根据未嵌套的版本说明了我想要的最终结果 (df_wider)。我的问题是：如何使用 purrr 从嵌套版本 (nested_df) 获得相同的最终结果 (df_wider)？

library(tidyverse)

df <- tibble(id_01 = c(rep("01", 3), rep("02", 3)),
             a = (c("a", "a", "b", "c", "c", "d")),
             b = letters[7:12],
             id_02 = rep(c(1, 2, 1), 2)
             )

df_wider <- pivot_wider(df,
                        id_cols = c(id_01, a),
                        names_from = id_02,
                        values_from = b,
                        names_sep = "_"
)

nested_df <- nest(df, data = -id_01)

为了清楚起见，我试图在数据框嵌套时进行旋转（即在取消嵌套之前）。

【问题讨论】：

您是否坚持使用purrr 来解决这个问题？因为使用unnest 会简单得多。
是的，我需要使用 purrr，因为实际的数据帧有很多 GB，并且在取消嵌套后旋转会耗尽 R 的内存。

标签： r tidyverse purrr

【解决方案1】：

我们可以在dplyr::mutate() 中使用purrr::map()：

library(tidyverse)

df <- tibble(
  id_01 = c(rep("01", 3), rep("02", 3)),
  a = (c("a", "a", "b", "c", "c", "d")),
  b = letters[7:12],
  id_02 = rep(c(1, 2, 1), 2)
)

nested_df <- df %>%
  nest(data = -id_01) %>% 
  mutate(data = map(data, ~ .x %>%
    pivot_wider(
      id_cols = a,
      names_from = id_02,
      values_from = b
    )))

nested_df
#> # A tibble: 2 x 2
#>   id_01 data            
#>   <chr> <list>          
#> 1 01    <tibble [2 x 3]>
#> 2 02    <tibble [2 x 3]>

nested_df %>% 
  unnest(data)
#> # A tibble: 4 x 4
#>   id_01 a     `1`   `2`  
#>   <chr> <chr> <chr> <chr>
#> 1 01    a     g     h    
#> 2 01    b     i     <NA> 
#> 3 02    c     j     k    
#> 4 02    d     l     <NA>

^{由reprex package (v1.0.0) 于 2021 年 3 月 26 日创建}

【讨论】：