Question

我正在寻找一种方法将字符串作为输入传递给我自己函数中dplyr包中的filter_函数。我已按如下方式设置数据框：

df = data.frame(
    X1 = LETTERS[1:5], 
    X2 = c("apple", "apple", "apple", "banana", "banana")
)

我正在寻找一种方法来编写一个函数，我可以在其中传递＆＃34; apple＆＃34;或＆＃34;香蕉＆＃34;过滤数据框。

我试过了：

filterFruit = function(Data, Fruit){
   retVal = filter_(Data, "X2 == Fruit")
   return(retVal)
}

然后传递值：

apple1 = filterFruit(df, "apple")
apple1

这会返回一个错误：

Error: object 'Fruit' not found

我尝试了其他一些方法，但没有成功，我希望有人可以提供帮助。

修改

我已经意识到我不需要使用filter_来进行此操作，因为我没有选择我要过滤的列，并且可以将参数传递给没有引号的过滤器。然而，问题仍然存在于：

df = data.frame(
    X1 = LETTERS[1:5], 
    X2 = c("apple", "apple", "apple", "banana", "banana")
    X3 = c("apple", "banana", "apple", banana", "apple")
)

并且需要决定您需要过滤哪个列（X2或X3）。

Answer 1

这应该回答你的编辑：

library(dplyr)

df = data.frame(
  X1 = LETTERS[1:5], 
  X2 = c("apple", "apple", "apple", "banana", "banana"),
  X3 = c("apple", "banana", "apple", "banana", "apple"), 
  stringsAsFactors=FALSE
)

column_string = "X2"
column_value = "banana"
column_name <- rlang::sym(column_string)

filtered_df <- df %>%
  filter(UQ(column_name) == UQ(column_value))

filtered_df

Answer 2

我已经使用which来获得成果。要获得索引，您可以使用retVal。

filterFruit <-  function(Data, column, Fruit){
  idx <- Data[,column]
  retVal <-  which(idx == Fruit)
  fruits_here <- Data[c(retVal), column]
  return(fruits_here)
}

如何将字符串传递给函数中的dplyr过滤器？

2 个答案: