Saya ingin membuat fungsi di mana input pertama create_df diubah menjadi nomor 1 menggunakan rapieval. Di sini seharusnya hanya kolom cyl.

Bagaimana cara menarik "input" pertama ...?

  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

Dibuat pada 07-01-2020 oleh paket reprex (v0.3.0)

0
John-Henry 1 Juli 2020, 18:57

1 menjawab

Jawaban Terbaik

Anda dapat menarik variabel pertama keluar dari titik-titik dengan hanya mengambil elemen pertama dari vars (dengan kredit untuk Lionel Henry untuk menunjukkan hal ini).

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
3
Allan Cameron 1 Juli 2020, 16:22