2017-07-03 53 views
7

我想写一个自定义函数,将基于预定义的变量向量(例如,vector_heavy)的值计算新变量,然后根据参数命名新变量提供给函数(例如,custom_name)。基于一个静态命名一个新的变量

这个变量命名是我的平静技能失败的地方。任何帮助是极大的赞赏。

library(tidyverse) 

vector_heavy <- quos(disp, wt, cyl) 

cv_compute <- function(data, cv_name, cv_vector){ 
    cv_name <- enquo(cv_name) 
    data %>% 
    rowwise() %>% 
    mutate(!!cv_name = mean(c(!!!cv_vector), na.rm = TRUE)) %>% 
    ungroup() 
} 

d <- cv_compute(mtcars, cv_name = custom_name, cv_vector = vector_heavy) 

我的错误信息上写着:

Error: unexpected '=' in: 
" rowwise() %>% 
    mutate(!!cv_name =" 

cv_name之前卸下!!mutate()将导致计算出一个新的变量命名字面上一个cv_name功能,而忽略我已经包括作为该custom_name论据。

cv_compute <- function(data, cv_name, cv_vector){ 
    cv_name <- enquo(cv_name) 
    data %>% 
    rowwise() %>% 
    mutate(cv_name = mean(c(!!!cv_vector), na.rm = TRUE)) %>% 
    ungroup() 
} 

我怎样才能得到这个功能利用我公司供应作为cv_name一个参数custom_name

回答

7

您需要在mutate内使用:=帮手。您还需要quo_name将输入转换为字符串。

mutate线的功能,然后将看起来像

mutate(!!quo_name(cv_name) := mean(c(!!!cv_vector), na.rm = TRUE))

在其整体:

cv_compute <- function(data, cv_name, cv_vector){ 
    cv_name <- enquo(cv_name) 
    data %>% 
      rowwise() %>% 
      mutate(!!quo_name(cv_name) := mean(c(!!!cv_vector), na.rm = TRUE)) %>% 
      ungroup() 
} 

cv_compute(mtcars, cv_name = custom_name, cv_vector = vector_heavy) 

    mpg cyl disp hp drat wt qsec vs am gear carb custom_name 
    <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>  <dbl> 
1 21.0  6 160 110 3.90 2.620 16.46  0  1  4  4 56.20667 
2 21.0  6 160 110 3.90 2.875 17.02  0  1  4  4 56.29167 
3 22.8  4 108 93 3.85 2.320 18.61  1  1  4  1 38.10667 
4 21.4  6 258 110 3.08 3.215 19.44  1  0  3  1 89.07167 
5 18.7  8 360 175 3.15 3.440 17.02  0  0  3  2 123.81333 
6 18.1  6 225 105 2.76 3.460 20.22  1  0  3  1 78.15333 
+0

漂亮的解决方案谢谢! – Joe