自定义函数中的Tidyselect

Question

我想使用var1:var20 tidyselect语法创建一个自定义函数，该函数可以应用于数据帧中的一系列变量，但我遇到了一个错误。下面是一个基本的例子：

library(dplyr)
mtcars
test_func <- function(start_var,end_var) {
  mutate(across(start_var:end_var, ~ifelse(.x!=0,.x+1,.x)))
}
mtcars %>% test_func(mpg,hp)

预期的结果(不使用该函数)是：

mtcars %>% 
  mutate(across(mpg:hp, ~ifelse(.x!=0,.x+1,.x)))

使用该函数将导致以下错误消息：

error in test_func(., mpg, hp) : unused argument (hp)

Answer 1

以下是一个潜在的解决方案：

library(dplyr)

test_func <- function(data, start_var, end_var) {
  data %>% 
    mutate(across({{start_var}}:{{end_var}}, ~ifelse(.x!=0,.x+1000,.x)))
}

mtcars %>% 
  test_func(mpg, hp)

我将您的“其他”arg更改为1000，以突出显示该更改：

                     mpg  cyl disp   hp drat    wt  qsec vs am gear carb
Mazda RX4         1021.0 1006 1160 1110 3.90 2.620 16.46  0  1    4    4
Mazda RX4 Wag     1021.0 1006 1160 1110 3.90 2.875 17.02  0  1    4    4
Datsun 710        1022.8 1004 1108 1093 3.85 2.320 18.61  1  1    4    1
Hornet 4 Drive    1021.4 1006 1258 1110 3.08 3.215 19.44  1  0    3    1
Hornet Sportabout 1018.7 1008 1360 1175 3.15 3.440 17.02  0  0    3    2

更多关于为什么可以找到这里的信息

Answer 2

您可以对您的函数进行以下调整以使其工作：

test_func <- function(data, start_var, end_var) { # add data source to the function
  data %>% mutate(across({{start_var}}:{{end_var}}, ~ifelse(.x!=0,.x+1,.x))) # add {{}}
}

test_func(mtcars, mpg, hp)