R中的输入变量名称处理

Question

我想了解如何处理函数中变量名的输入。比如我写了一个这样的函数：

bin_multi <- function(y, dataframe, sel = NULL){
  if(!is.null(sel)) {
     dataframe <- dataframe[,sel]}
  else {
     dataframe <- dataframe[!y]}
}

其中dataframe 是输入数据帧，y 是数据帧中的目标变量，sel 是从dataframe 中选择的列，例如sel = c(1,2,3)。

此函数的目的是简单地获取具有给定sel 的dataframe 的子集，并且当未给出sel 时，从dataframe 中排除y 目标变量。

我的问题是，我怎样才能在这个函数中正确引用y？在输入中，y 是变量的名称。 deparse() 能解决这个问题吗？

谢谢大家。

Answer 1

我认为这会奏效：

bin_multi <- function(y, dataframe, sel = NULL){

  if(!is.null(sel)) {

     dataframe <- dataframe[,sel]

  } else {

     dataframe <- dataframe[,which(names(dataframe) != deparse(substitute(y)))]

  }

}

这是利用this answer 将您的对象名称转换为字符串。