R 对数据表中的`i`求值.S`set()`

发布于11月11日

假设我在一个data.table个对象中创建一个新列:

require(data.table)
data(iris)
dt.iris <- data.table(iris)

dt.iris[,shortSpecies:=substr(Species,1,5)]

现在，我不想直接使用:=来执行此操作，而是希望有一个函数，该函数将创建列的代码作为参数，然后对其求值.我最终得到了这样的结果:

make_new_col <- function(inputDT, newColName, construction){
  set(
    x = inputDT, 
    j = newColName, 
    value=eval(expr = parse(text = construction), envir=inputDT)
    )
}

dt.iris <- make_new_col(
  inputDT = dt.iris, 
  newColName = 'shortSpecies', 
  construction = 'substr(Species,1,5)'
)

这是可行的，但现在我想添加一个条件，即使其等同于dt.iris[Sepal.Length>5,shortSpecies:=substr(Species,1,5)].我需要以某种方式将条件传递到set()的i=部分进行判断，但我找不到有效的解决方案.

推荐答案

您可以提供一个标准条件if cond是missing.

> library(data.table)
> make_new_col3 <- function(inputDT, cond, newColName, construction) {
+   stopifnot(ncol(inputDT) > 0)
+   if (missing(cond)) cond <- "`mode<-`(inputDT[[1]], 'logical')"
+   set(
+     x=inputDT, 
+     i=(w <- which(eval(parse(text=cond), envir=inputDT))),
+     j=newColName, 
+     value=eval(parse(text=construction), envir=inputDT[w, ])
+   )
+ }

> dt.iris <- make_new_col3(
+   inputDT=dt.iris, 
+   cond='Sepal.Length > 5',
+   newColName='shortSpecies', 
+   construction='substr(Species, 1, 5)'
+ )
> head(dt.iris)
   Sepal.Length Sepal.Width Petal.Length Petal.Width Species shortSpecies
1:          5.1         3.5          1.4         0.2  setosa        setos
2:          4.9         3.0          1.4         0.2  setosa         <NA>
3:          4.7         3.2          1.3         0.2  setosa         <NA>
4:          4.6         3.1          1.5         0.2  setosa         <NA>
5:          5.0         3.6          1.4         0.2  setosa         <NA>
6:          5.4         3.9          1.7         0.4  setosa        setos

> dt.iris <- make_new_col3(
+   inputDT=dt.iris, 
+   # cond='Sepal.Length > 5',
+   newColName='shortSpecies', 
+   construction='substr(Species, 1, 5)'
+ )
> head(dt.iris)
   Sepal.Length Sepal.Width Petal.Length Petal.Width Species shortSpecies
1:          5.1         3.5          1.4         0.2  setosa        setos
2:          4.9         3.0          1.4         0.2  setosa        setos
3:          4.7         3.2          1.3         0.2  setosa        setos
4:          4.6         3.1          1.5         0.2  setosa        setos
5:          5.0         3.6          1.4         0.2  setosa        setos
6:          5.4         3.9          1.7         0.4  setosa        setos

Data:个