将方程作为参数传递给函数

Question

有人告诉我，将方程作为字符串传递并评估它们是不好的做法。我怎样才能创建一个函数，它接受一个方程并在没有字符串版本且不使用第三方包的情况下对其进行评估？

这是我的功能：

replaceFormula <- function(df, column, formula){
  df[column] <- eval(parse(text=formula), df)
  return(df)
}

这是我的用例：

set.seed(24)
dataset <- matrix(sample(c(NA, 1:5), 25, replace = TRUE), 5)
df <- as.data.frame(dataset)
replaceFormula(df, 'V5', 'V3+V4')

更新：

这也有条件吗？ 我的示例函数：

replaceFactor <- function(df, column, condition, what){
  df[column] <- sapply(df[column],function(x) ifelse(eval(parse(text=condition), df), what, x))
  return(df)
}

我的用例：

set.seed(24)
dataset <- matrix(sample(c(NA, 1:5), 25, replace = TRUE), 5)
df <- as.data.frame(dataset)
replaceFactor(df, 'V5', 'V1==1', 'GOOD')

Answer 1

看起来你已经为自己制作了一个 transform 的笨拙版本

> set.seed(24)
> dataset <- matrix(sample(c(NA, 1:5), 25, replace = TRUE), 5)
> df <- as.data.frame(dataset)
> transform(df, V5 = V3 + V4)
  V1 V2 V3 V4 V5
1  1  5  3  5  8
2  1  1  2  1  3
3  4  4  4 NA NA
4  3  4  4  3  7
5  3  1  1 NA NA

Answer 2

我们可以将公式作为quosure 传递，并通过在dplyr 的开发版（或即将发布的0.6.0）中取消引用（!! 或UQ）来评估它

library(dplyr)
replaceFormula <- function(dat, Col, form){
      Col <- quo_name(enquo(Col))
      dat %>% 
          mutate(UQ(Col) := UQ(form))
  }


replaceFormula(df, V5, quo(V3 + V4)) 
#   V1 V2 V3 V4 V5
#1  1  5  3  5  8
#2  1  1  2  1  3
#3  4  4  4 NA NA
#4  3  4  4  3  7
#5  3  1  1 NA NA

更新

基于 OP 的 cmets，我们还可以传递一个表达式来评估和更改基于它的值

replaceFormulaNew <- function(dat, Col, form, what){
      Col <- enquo(Col)
      ColN <- quo_name(Col)
      what <- quo_name(enquo(what))


      dat %>% 
          mutate(UQ(ColN) := ifelse(UQ(form), what, UQ(Col)))
  }


replaceFormulaNew(df, V5, quo(V1==1), GOOD)
#   V1 V2 V3 V4   V5
#1  1  5  3  5 GOOD
#2  1  1  2  1 GOOD
#3  4  4  4 NA    4
#4  3  4  4  3 <NA>
#5  3  1  1 NA    1

replaceFormulaNew(df, V5, quo(V3 < V4), GOOD)
#   V1 V2 V3 V4   V5
#1  1  5  3  5 GOOD
#2  1  1  2  1    3
#3  4  4  4 NA <NA>
#4  3  4  4  3 <NA>
#5  3  1  1 NA <NA>

enquo 接受输入参数并将其转换为quosure，而quo_name 将其转换为字符串以在mutate 中进行评估，以将评估的输出分配给输入中指定的列