Напишите функцию tidyeval для переименования факторного уровня в dplyr

Question

Я пытаюсь написать функцию tidyeval, которая принимает числовой столбец, заменяет значения выше определенного limit значением для limit, превращает этот столбец в фактор и затем заменяет уровень фактора, равный limit с уровнем под названием "limit +".

Например, я пытаюсь заменить любое значение выше 3 в sepal.width на 3, а затем переименовать этот фактор в 3+.

В качестве примера, вот как я пытаюсь заставить его работать с набором данных iris.Однако функция fct_recode () неправильно переименовывает уровень фактора.

plot_hist <- function(x, col, limit) {
  col_enq <- enquo(col)
  x %>% 
    mutate(var = factor(ifelse(!!col_enq > limit, limit,!!col_enq)),
           var = fct_recode(var, assign(paste(limit,"+", sep = ""), paste(limit))))
}

plot_hist(iris, Sepal.Width, 3)

Answer 1

Чтобы исправить последнюю строку, мы можем использовать специальный символ :=, так как нам нужно установить значение в левой части выражения.Для RHS нам нужно привести к символу, поскольку fct_recode ожидает символьный вектор справа.

library(tidyverse)


plot_hist <- function(x, col, limit) {
  col_enq <- enquo(col)

  x %>% 
    mutate(var = factor(ifelse(!!col_enq > limit, limit, !!col_enq)),
           var = fct_recode(var, !!paste0(limit, "+") := as.character(limit)))
}

plot_hist(iris, Sepal.Width, 3) %>% 
  sample_n(10)
#>     Sepal.Length Sepal.Width Petal.Length Petal.Width    Species var
#> 40           5.1         3.4          1.5         0.2     setosa  3+
#> 98           6.2         2.9          4.3         1.3 versicolor 2.9
#> 7            4.6         3.4          1.4         0.3     setosa  3+
#> 99           5.1         2.5          3.0         1.1 versicolor 2.5
#> 76           6.6         3.0          4.4         1.4 versicolor  3+
#> 77           6.8         2.8          4.8         1.4 versicolor 2.8
#> 85           5.4         3.0          4.5         1.5 versicolor  3+
#> 119          7.7         2.6          6.9         2.3  virginica 2.6
#> 110          7.2         3.6          6.1         2.5  virginica  3+
#> 103          7.1         3.0          5.9         2.1  virginica  3+