R: Group_by 取決於條件

Question

假設我有這個簡單的數據框：

df <- tibble(a = c(1, 1), b = c(2, 2))

我現在想知道如何在依賴於變量的管道中使用group_by 。 就像是，

flag <- T
resulting <- df %>% 
              filter(a > 0 & b >0) %>%
              group_by(ifelse(flag), yes = c(a), no = c(a, b))

也就是說，如果flag == T ，那么我只想在a列上分組。 如果標志為false ，我想對兩列進行分組。

Answer 1

我認為這對我有用

flag <- T
resulting <- df %>% 
    filter(a > 0 & b >0) %>%
    {if(flag) group_by(.,a) else group_by(. ,a , b)}

resulting

# A tibble: 2 × 2
# Groups:   a [1]  # <======== here grouped by a
      a     b
  <dbl> <dbl>
1     1     2
2     1     2

通過更改標志

flag <- F
resulting <- df %>% 
    filter(a > 0 & b >0) %>% 
    {if(flag) group_by(.,a) else group_by(. ,a , b)}

resulting

# A tibble: 2 × 2
# Groups:   a, b [1]   # <======== here grouped by a ,b
      a     b
  <dbl> <dbl>
1     1     2
2     1     2