将列名作为参数传递给函数

Question

我创建了一个数据表df

library(data.table)
df <- data.table(id = c(1,1,1,2,2), starts = c(0,0,6,0,9), ends = c(0,6,10,9,20))

df
   #id starts ends
#1:  1      0    0
#2:  1      0    6
#3:  1      6   10
#4:  2      0    9
#5:  2      9   20

我定义了一个函数＆＃39; equal_0＆＃39;，即输入col_name，然后该函数将返回一个数据表col_name == 0

equal_0 <- function(col_name){
    a <- df[col_name == 0]
    return(a)
}

例如，equal_0(starts)等于df[starts == 0]，预期结果为：

df[starts==0]
   #id starts ends
#1:  1      0    0
#2:  1      0    6
#3:  2      0    9

但是，equal_0(starts)和equal_0("starts")都无法正常工作。

错误消息是：

equal_0("starts")
#Empty data.table (0 rows) of 3 cols: id,starts,ends

equal_0(starts)
#Error in eval(.massagei(isub), x, parent.frame()) : object 'starts' not found 

如何将starts传递给函数equal_0？

Answer 1

使用get从data.table中提取列：

equal_0 <- function(col_name){
    library(data.table)
    a <- df[get(col_name) == 0]
    return(a)
}
equal_0("starts")
   id starts ends
1:  1      0    0
2:  1      0    6
3:  2      0    9