传递数据框以在函数内变异

Question

我想传递一个数据框及其列,以便在函数中由dplyr的mutate处理.

这是一个例子

multifun <- function(dataf,vari){
 mutate(dataf,newvar=vari*2)
}

multifun(mtcars,gear)

此功能的问题在于变量"gear"不是可识别的对象.更具体地说,我得到了错误

mutate_impl(.data,named_dots(...),environment())错误找不到对象'gear'

这是dplyr的mutate正在寻找有问题的变量的环境问题.

我明白那个

multifun(mtcars,mtcars$gear)

会给我一个我想要的答案,即

    mpg  cyl  disp  hp   drat  wt   qsec  vs am   gear carb newvar
1  21.0   6   160.0 110  3.90 2.620 16.46  0  1    4    4      8
2  21.0   6   160.0 110  3.90 2.875 17.02  0  1    4    4      8
3  22.8   4   108.0  93  3.85 2.320 18.61  1  1    4    1      8

但我想看看是否有办法避免在函数调用中引用数据框中使用的每个变量.

我也知道从函数调用中取出mutate没有问题.也就是说,mutate(mtcars,newvar=gear*2)这份工作.但是,我试图理解dplyr的mutate在置于函数内时如何在不同的环境中查找有问题的变量.

Answer 1

这对我来说确实很丑陋，但似乎可行。基本上，我尝试使用，get但是它似乎不知道在哪里寻找，因此我指定了环境。

multifun <- function(dataf, vari){
  vari <- deparse(substitute(vari))
  mutate(dataf, newvar = get(vari, envir = as.environment(dataf)) * 2)
}

输出：

multifun(mtcars, gear)
#                      mpg cyl  disp  hp drat    wt  qsec vs am gear carb newvar
# Mazda RX4           21.0   6 160.0 110 3.90 2.620 16.46  0  1    4    4      8
# Mazda RX4 Wag       21.0   6 160.0 110 3.90 2.875 17.02  0  1    4    4      8
# Datsun 710          22.8   4 108.0  93 3.85 2.320 18.61  1  1    4    1      8
# <<<SNIP>>>
# Ferrari Dino        19.7   6 145.0 175 3.62 2.770 15.50  0  1    5    6     10
# Maserati Bora       15.0   8 301.0 335 3.54 3.570 14.60  0  1    5    8     10
# Volvo 142E          21.4   4 121.0 109 4.11 2.780 18.60  1  1    4    2      8

Answer 2

查看@Anandas解决方案，这似乎是最简单的技巧

multifun <- function(dataf, vari){   
dataf <- mutate(dataf, newvar = dataf[, vari]*2);   
return(dataf) 
}

multifun(mtcars,"gear")

同样，考虑到@Anandas的建议，您也可以

multifun <- function(dataf, vari){  
  vari <- deparse(substitute(vari))
  dataf <- mutate(dataf, newvar = dataf[, vari]*2)   
  return(dataf) 
}

multifun(mtcars, gear)