使用 dplyr / magrittr 传递分组的 data.frame答案

【问题标题】：Pass grouped data.frame using dplyr / magrittr使用 dplyr / magrittr 传递分组的 data.frame
【发布时间】：2020-10-18 11:33:31
【问题描述】：

使用base::by() 和data.table，我们可以按变量分组，然后能够访问分组子集的data.frame。我怎样才能用magrittr 或dplyr 做同样的事情？

我尝试了tib %>% group_by(grp) %>% mutate(V2 = fx(.))，但不是通过子组，而是通过 LHS 中的整个分组 tibble。这是一个 MRE：

library(dplyr)
tib = tibble(grp = rep(1:2, 1:2),
             V1 = 1:3)
tib
#> # A tibble: 3 x 2
#>     grp    V1
#>   <int> <int>
#> 1     1     1
#> 2     2     2
#> 3     2     3

fx = function(x){
  ans = seq(nrow(x))
  print(ans)
}

tib %>%
  group_by(grp)%>%
  mutate(V2 = fx(.))
#> [1] 1 2 3
#> Error: Problem with `mutate()` input `V2`.
#> x Input `V2` can't be recycled to size 1.
#> i Input `V2` is `fx(.)`.
#> i Input `V2` must be size 1, not 3.
#> i The error occured in group 1: grp = 1.

这是我希望使用data.table 的行为：

library(data.table)
as.data.table(tib)[, V2 := fx(.SD), grp][]
#> [1] 1
#> [1] 1 2
#>      grp    V1    V2
#>    <int> <int> <int>
#> 1:     1     1     1
#> 2:     2     2     1
#> 3:     2     3     2

【问题讨论】：

标签： magrittr dplyr data.table r dplyr magrittr

【解决方案1】：

从dplyr 1.0.0 起，您可以使用cur_data()。

library(dplyr)
tib %>% group_by(grp)%>% mutate(V2 = fx(cur_data()))

#[1] 1
#[1] 1 2
# A tibble: 3 x 3
# Groups:   grp [2]
#    grp    V1    V2
#  <int> <int> <int>
#1     1     1     1
#2     2     2     1
#3     2     3     2

请注意，cur_data() 传递数据时没有分组变量 (grp)。如果您想将分组变量传递给函数，请改用cur_data_all()。

【讨论】：