将列名作为参数传递给 data.table::setkey() --- 有些列不在 data.table 中：col_name

Question

所以，基本上我想要的类似于这两篇文章： 1 , 2 。 但是，当我尝试解决方案时，我不断收到错误消息。

我的问题的不同之处在于我正在使用 data.table 包并试图设置一个键值。 有关更多详细信息，请参阅此处。

现在，举个例子，让我们假设我有一个数据框并设置它的键如下：

data <- data.table::as.data.table(data.frame(A = c(1, 2, 3), B = c("one", "two", "three")))
 data <- data.table::setkey(data, A)

这有效。 现在，我可以通过其他一些数据结构进行过滤，如下所示：

matches <- data[c(1)]

上面的行将创建一个 data.table，它是data的子集，其中变量A值为 1。

现在，让我们假设我想让它成为一个通用函数。 我无法让以下工作：

genericFunction <- function(data, col_name, filter){
    #Convert data.frame to data.table
    data <- data.table::as.data.table(data)

    #Set the key based on a variable name
    #Error is in this step
    data <- data.table::setkey(data, col_name)

    #Save the subset of data
    matches <- data[c(sorter)]

    return(matches)
}

也就是说，如果我去执行以下操作：

exampleData <- data.frame(A = c(1, 2, 3), B = c("one", "two", "three"))
exampleName <- "A"
exampleFilter <- 1

genericFunction(exampleData, exampleName, exampleFilter)

我收到以下错误：

 Error in setkeyv(x, cols, verbose = verbose, physical = physical) : 
  some columns are not in the data.table: col_name 

我知道我想使用 lazyeval::interp() 或类似的东西，但是，上面示例链接中的实现对我不起作用。 有没有人对我应该做什么有任何想法？ 任何帮助表示赞赏。

Answer 1

不是 data.table 专家，而是?setkey说：

setkey(x, ..., verbose=getOption("datatable.verbose"), physical = TRUE)

... - The columns to sort by. Do not quote the column names.

这意味着您不能在此处传递带引号的列名。

您可以使用setkeyv ：

setkeyv(x, cols, verbose=getOption("datatable.verbose"), physical = TRUE)

cols - A character vector of column names

genericFunction <- function(data, col_name, filter){
  #Convert data.frame to data.table
  data <- data.table::as.data.table(data)
  
  data <- data.table::setkeyv(data, col_name)
  
  #Save the subset of data
  matches <- data[c(filter)]
  
  return(matches)
}

exampleData <- data.frame(A = c(1, 2, 3), B = c("one", "two", "three"))
exampleName <- "A"
exampleFilter <- 1

genericFunction(exampleData, exampleName, exampleFilter)

#   A   B
#1: 1 one