连接quosures和字符串

时间:2017-09-13 20:54:10

标签: r dplyr rlang tidyeval

我正在寻找一种方法来连接一个quosure和一个结果为quosure的字符串。实际上,如果我使用paste0()quo_name(),我就可以做到。但是我想知道在我的包中编写一个函数是否有更优雅的选择。这是一个通用的例子:

library(dplyr)

df <- data_frame(
  z_1 = 1,
  z_2 = 2,
  y_1 = 10,
  y_2 = 20
)

get_var <- function(.data, var) {
  xx = enquo(var)

  select(.data, paste0(quo_name(xx), "_1"), paste0(quo_name(xx), "_2"))
}

get_var(df, z)
# A tibble: 1 x 2
    z_1   z_2
  <dbl> <dbl>
1     1     2

1 个答案:

答案 0 :(得分:3)

如果没有功能,您可以使用dplyr

进行操作
library(dplyr)
df %>%
  select(starts_with("z_"))

您还可以创建一个函数并为变量名称传递一个字符串,如下所示:

get_var= function(df, var){
  df %>%
    select(starts_with(paste0(var, "_")))
}

get_var(df, "z")

现在,当您尝试传入变量名而不将其引用到函数(R代码,而不是它包含的值)时,会出现棘手的部分。一种方法是在基地R中deparse + substitute。这会将提供给var符号转换为字符串,方便以后使用在函数内:

get_var = function(df, var){
  var_quo = deparse(substitute(var))
  df %>%
    select(starts_with(paste0(var_quo, "_")))
}

最后,以下是如何对enquo包中的quo_namerlang/tidyverse执行相同操作:

library(rlang)
get_var = function(df, var){
  var_quo = quo_name(enquo(var))
  df %>%
    select(starts_with(paste0(var_quo, "_")))
}

get_var(df, z)
get_var(df, y)

<强>结果:

# A tibble: 1 x 2
    z_1   z_2
  <dbl> <dbl>
1     1     2

# A tibble: 1 × 2
    y_1   y_2
  <dbl> <dbl>
1    10    20

备注:

  1. quosures是引用表达式,用于跟踪环境。
  2. enquo使用一个引用函数参数的符号,引用R代码并将其与函数的环境捆绑在一起。
  3. quo_name将一个quosure格式化为一个字符串,以后可以在函数中使用。
  4. quo_textquo_name类似,但不会检查输入是否为符号。
  5. 检查这些:

    1. rlang documentation
    2. Non-Standard Evaluation in R
    3. ?enquo
    4. ?quo_name