Символьный параметр в переменной

Question

Я хочу определить функцию

factor <- function(var1)
  {
     df['var1'] <- as.factor(df['var1'])
  }

для преобразования типа переменной

и используйте вот так factor(var1) без кавычек!

Answer 1

Используя подход tidyeval, вы можете использовать rlang::sym для преобразования строки в символ, а затем заключить ее в кавычки внутри mutate с помощью !!. Смотри также этот ответ .

library(rlang)
library(tidyverse)

# create a test data frame
iris1 <- head(iris) %>% 
  as.tibble() %>% 
  mutate(Species = as.character(Species))
iris1

#> # A tibble: 6 x 5
#>   Sepal.Length Sepal.Width Petal.Length Petal.Width Species
#>          <dbl>       <dbl>        <dbl>       <dbl> <chr>  
#> 1          5.1         3.5          1.4         0.2 setosa 
#> 2          4.9         3            1.4         0.2 setosa 
#> 3          4.7         3.2          1.3         0.2 setosa 
#> 4          4.6         3.1          1.5         0.2 setosa 
#> 5          5           3.6          1.4         0.2 setosa 
#> 6          5.4         3.9          1.7         0.4 setosa

str(iris1)

#> Classes 'tbl_df', 'tbl' and 'data.frame':    6 obs. of  5 variables:
#>  $ Sepal.Length: num  5.1 4.9 4.7 4.6 5 5.4
#>  $ Sepal.Width : num  3.5 3 3.2 3.1 3.6 3.9
#>  $ Petal.Length: num  1.4 1.4 1.3 1.5 1.4 1.7
#>  $ Petal.Width : num  0.2 0.2 0.2 0.2 0.2 0.4
#>  $ Species     : chr  "setosa" "setosa" "setosa" "setosa" ...

# build function
myfactor <- function(df, var1)
{
  var1 <- sym(var1)
  df <- df %>% mutate(!! var1 := as.factor(!! var1))
  return(df)
}

myfactor(iris1, 'Species')

#> # A tibble: 6 x 5
#>   Sepal.Length Sepal.Width Petal.Length Petal.Width Species
#>          <dbl>       <dbl>        <dbl>       <dbl> <fct>  
#> 1          5.1         3.5          1.4         0.2 setosa 
#> 2          4.9         3            1.4         0.2 setosa 
#> 3          4.7         3.2          1.3         0.2 setosa 
#> 4          4.6         3.1          1.5         0.2 setosa 
#> 5          5           3.6          1.4         0.2 setosa 
#> 6          5.4         3.9          1.7         0.4 setosa

Создано в 2018-07-02 пакетом представ. (v0.2.0.9000).

Answer 2

вам нужно использовать substitute для захвата имени переменной, а затем сделать имя / символ deparse

facr <- function(var1)
{
  v=deparse(substitute(var1))
  df[v] <- as.factor(df[[v]])
  df
}