R，使用 dplyr::filter() 和 %in% 将列名作为参数传递给函数

Question

如何在类似于问题 here 的函数中传递列名但使用 dplyr链接和 filter()连同%in% .

require(dplyr)
set.seed(8)
df <- data.frame(
  A=sample(c(1:3), 10, replace=T), 
  B=sample(c(1:3), 10, replace=T))

如果想获取 A 列为 1 或 2 的行，我可以这样做:

df %>% filter(A %in% c(1,2))

我得到:

  A B
1 2 3
2 1 2
3 1 3
4 2 1
5 1 1
6 1 3

现在，我怎么能把它放在一个函数中，在那里可以指定列，这不起作用:

fun1 <- function(x, column, n){
  res <- 
    x %>% filter(column %in% n)
  return(res)
}
fun1(df, A, c(1,2))

Answer 1

你可以试试

fun1 <- function(x, column, n){
 x %>% 
  filter_(lazyeval::interp(quote(x %in% y), x=as.name(column), y=n))
 }
fun1(df, 'A', 1:2)

或者

fun2 <- function(x, column, n){
   args <- as.list(match.call())
   x %>%
     filter(eval(args$column, x) %in% n)
 }

 fun2(df, A, 1:2)