понимание языка: мутировать с именем переменной col и столбцом переменной - PullRequest
Новая
       32

понимание языка: мутировать с именем переменной col и столбцом переменной

2 голосов
/ 10 марта 2019

Я бы хотел определить функцию, которая принимает data.frame и имя столбца и возвращает data.frame с преобразованным столбцом (например, в нижний регистр). Когда имя столбца известно заранее, это просто: 

diamonds %>% mutate(cut = tolower(cut))

Как определить функцию foo, такую, что: 

col <- "cut"
foo(diamonds, col)

это поведение такое же? (Не ищу базового ответа R или data.table, поскольку я хочу сохранить возможность dplyr перевести это в SQL-вызов с ленивой оценкой).

Если бы я просто хотел, чтобы моя функция работала с использованием: foo(diamonds, cut), мне просто нужны enquo и !!

foo <- function(df, col){
    x <- enquo(col)
    mutate(df, !!x := tolower(!!x))
}

Если я хочу взять имя столбца в кавычках, foo(diamonds, "cut"), достаточно добавить ensym: 

foo <- function(df, col){
    col <- ensym(col)
    x <- enquo(col)
    mutate(df, !!x := tolower(!!x))
}

но происходит сбой, когда передается переменная для аргумента: 

col <- "cut"
foo(diamonds, col) 

Error in ~col : object 'col' not found

Чего мне не хватает, чтобы оценить переменную?

Ответы [ 3 ]

7 голосов
/ 10 марта 2019

Вы также можете полностью избежать аккуратного вычисления, используя mutate_at(). 

library(tidyverse)

(x <- tibble(
  num = 1:3,
  month = month.abb[num]
))
#> # A tibble: 3 x 2
#>     num month
#>   <int> <chr>
#> 1     1 Jan  
#> 2     2 Feb  
#> 3     3 Mar

x %>%
  mutate(month = tolower(month))
#> # A tibble: 3 x 2
#>     num month
#>   <int> <chr>
#> 1     1 jan  
#> 2     2 feb  
#> 3     3 mar

foo <- function(df, col) {
  mutate_at(df, .vars = col, .funs = tolower)
}

foo(x, "month")
#> # A tibble: 3 x 2
#>     num month
#>   <int> <chr>
#> 1     1 jan  
#> 2     2 feb  
#> 3     3 mar

this <- "month"
foo(x, this)
#> # A tibble: 3 x 2
#>     num month
#>   <int> <chr>
#> 1     1 jan  
#> 2     2 feb  
#> 3     3 mar

Создано в 2019-03-09 пакетом Представить (v0.2.1.9000)

2 голосов
/ 10 марта 2019 
library(tidyverse)

col <- "cut"

foo <- function(df, col) {
  df %>% 
    mutate(!!sym(col) := tolower(!!sym(col)))
}

foo(diamonds, col)

Извлечь Передать строку как имя переменной в dplyr :: filter .

0 голосов
/ 10 марта 2019

Возвращаясь к исходному примеру, просто используйте ensym() для преобразования текстовых аргументов в символы, в этом случае нет необходимости в выражении. 

library(ggplot2)
col <- "cut"
foo <- function(df, col){
    col <- rlang::sym(col)
    dplyr::mutate(df, !!col := tolower(!!col))
}

foo(diamonds, col)
#> # A tibble: 53,940 x 10
#>    carat cut       color clarity depth table price     x     y     z
#>    <dbl> <chr>     <ord> <ord>   <dbl> <dbl> <int> <dbl> <dbl> <dbl>
#>  1 0.23  ideal     E     SI2      61.5    55   326  3.95  3.98  2.43
#>  2 0.21  premium   E     SI1      59.8    61   326  3.89  3.84  2.31
#>  3 0.23  good      E     VS1      56.9    65   327  4.05  4.07  2.31
#>  4 0.290 premium   I     VS2      62.4    58   334  4.2   4.23  2.63
#>  5 0.31  good      J     SI2      63.3    58   335  4.34  4.35  2.75
#>  6 0.24  very good J     VVS2     62.8    57   336  3.94  3.96  2.48
#>  7 0.24  very good I     VVS1     62.3    57   336  3.95  3.98  2.47
#>  8 0.26  very good H     SI1      61.9    55   337  4.07  4.11  2.53
#>  9 0.22  fair      E     VS2      65.1    61   337  3.87  3.78  2.49
#> 10 0.23  very good H     VS1      59.4    61   338  4     4.05  2.39
#> # … with 53,930 more rows

Создано в 2019-03-11 * представит пакет (v0.2.1)

...