如何通过将字符串作为用户定义函数中的列名传递为列名来过滤？

Question

我正在编写一个函数，其中用户指定要过滤的列以及所需的截止值。在此示例中，我想滤除2下的任何预测分数。这是一个示例数据集：

library(dplyr)

test <- tibble(name = c("Corey", "Justin", "Sibley", "Kate"),
               pretest_score = c(1:4),
               posttest_score = c(5:8),
               final_score = c(9:12))


filter_function <- function(data, test_type = c(pretest, posttest, final), value) {
  
  test_character <- deparse(substitute(test_type))
  test_score <- paste0(test_character, "_score")
  
  data %>%
    filter({{test_score}} > value)
  
}

filter_function(test, test_type = pretest, value = 2)

我也尝试了！test_score，test_score（周围什么都没有），以及rlang ，无济于事。

注意：我知道在 this 示例中，我只能将Pretest_score，tostTest_score等指定为测试类型，但是在我的真实数据集中，我对这些测试有许多维度，用户可以确定这些测试的临界值（ Pretest_score，Pretest_date，Pretest_location等），因此，我必须在功能本身中与后缀（此处，_score）合并列前缀。

感谢您的任何帮助！

Answer 1

将字符转换为sym bol并使用!!进行评估

filter_function <- function(data, test_type = c(pretest, posttest, 
      final), value) {
  
  test_character <- deparse(substitute(test_type))
  test_score <- paste0(test_character, "_score")
  
  data %>%
    filter(!! rlang::sym(test_score) > value)
  
}

- 测试

> filter_function(test, test_type = pretest, value = 2)
# A tibble: 2 × 4
  name   pretest_score posttest_score final_score
  <chr>          <int>          <int>       <int>
1 Sibley             3              7          11
2 Kate               4              8          12

Answer 2

几点：

• 通常，当R中使用一组选项时，一个使用字符向量而不是未评估的表达式。 r专门为此目的提供match.arg。这还实现了第一个选项的默认值，因此，如果我们使用match.arg调用函数可以省略type_test =“ Prepest”，因为这是默认值。 /p>

• 。[[test_score]]可以用来指定指示的列

因此我们

filter_function <- function(data, test_type = c("pretest", "posttest", "final"), 
  value) {

  test_type <- match.arg(test_type)
  test_score <- paste0(test_type, "_score")
  
  data %>%
    filter(.[[test_score]] > value)
  
}

filter_function(test, test_type = "pretest", value = 2)
# A tibble: 2 x 4
  name   pretest_score posttest_score final_score
  <chr>          <int>          <int>       <int>
1 Sibley             3              7          11
2 Kate               4              8          12

# pretest is the default
filter_function(test, value = 2)
# A tibble: 2 x 4
  name   pretest_score posttest_score final_score
  <chr>          <int>          <int>       <int>
1 Sibley             3              7          11
2 Kate               4              8          12

还指出我们可以像这样指定函数。由于match.arg，用户仍然可以指定“预测”。实际上，他们甚至可以指定“ pre”或“ post”。

filter_function2 <- function(data, 
    test_type = c("pretest_score", "posttest_score", "final_score"), 
    value) {

  test_type <- match.arg(test_type)
  
  data %>%
    filter(.[[test_type]] > value)
  
}

filter_function2(test, test_type = "pretest", value = 2)

基本r

这也可以在没有任何类似包的情况下完成：

filter_function3 <- function(data, test_type = c("pretest", "posttest", "final"), 
  value) {

  test_type <- match.arg(test_type)
  test_score <- paste0(test_type, "_score")
  
  data[data[[test_score]] > value, ]

}

filter_function3(test, test_type = "pretest", value = 2)