我怎样才能接受...的第一个输入？

Question

我想创建一个函数，使用 tidyeval 将 create_df 的第一个输入转换为数字 1。这里应该只有cyl列。

如何提取 ... 的第一个“输入”？

  library(dplyr, quietly = T)

create_df <- function(...){
  
  var <- enquos(...)
  first <- as_label(quos(...))
  
  mtcars %>% 
    group_by(!!!var) %>% 
    summarise(mean = mean(mpg)) %>% 
    mutate(!!first := 1)
}

create_df(cyl, am)
#> `summarise()` regrouping output by 'cyl' (override with `.groups` argument)
#> # A tibble: 6 x 4
#> # Groups:   cyl [3]
#>     cyl    am  mean `<quos>`
#>   <dbl> <dbl> <dbl>    <dbl>
#> 1     4     0  22.9        1
#> 2     4     1  28.1        1
#> 3     6     0  19.1        1
#> 4     6     1  20.6        1
#> 5     8     0  15.0        1
#> 6     8     1  15.4        1

^{由 reprex package (v0.3.0)}

于 2020-07-01 创建

Answer 1

您可以通过简单地从 vars 中取出第一个元素来从点中取出第一个变量（感谢 Lionel Henry 指出这一点）。

create_df <- function(...){
  
  var <- enquos(...)
  first <- as_label(var[[1]])
  
  mtcars %>% 
    group_by(!!!var) %>% 
    summarise(mean = mean(mpg)) %>% 
    mutate(!!first := 1)
}

create_df(cyl, am)
#> `summarise()` regrouping output by 'cyl' (override with `.groups` argument)
#> # A tibble: 6 x 3
#> # Groups:   cyl [1]
#>     cyl    am  mean
#>   <dbl> <dbl> <dbl>
#> 1     1     0  22.9
#> 2     1     1  28.1
#> 3     1     0  19.1
#> 4     1     1  20.6
#> 5     1     0  15.0
#> 6     1     1  15.4

我怎样才能接受...的第一个输入？

How can I take the first input of ...?

r

dplyr

tidyeval