r - 如何使用 dplyr 计算嵌套数据框中的行数

Question

这是一个愚蠢的示例数据框:

df <- data_frame(A = c(rep(1, 5), rep(2, 4)), B = 1:9) %>% 
  group_by(A) %>% 
  nest()

看起来像这样:

> df
# A tibble: 2 × 2
      A             data
  <dbl>           <list>
1     1 <tibble [5 × 1]>
2     2 <tibble [4 × 1]>

我想添加名为 N 的第三列条目等于 data 中每个嵌套数据帧中的行数.我想这会奏效:

> df %>% 
+   mutate(N = nrow(data))
Error: Unsupported type NILSXP for column "N"

怎么了？

Answer 1

合并dplyr和 purrr你可以这样做:

library(tidyverse)

df %>% 
  mutate(n = map_dbl(data, nrow))
#> # A tibble: 2 × 3
#>       A             data     n
#>   <dbl>           <list> <dbl>
#> 1     1 <tibble [5 × 1]>     5
#> 2     2 <tibble [4 × 1]>     4

我喜欢这种方法，因为您保持在通常的工作流程中，在 mutate 中创建一个新列，但利用了 map_* -family，因为你需要对列表进行操作。