在 R data.table 中，添加一个名称基于字符串的列

Question

我想在 data.table 中添加一列。但是这个新列的名称必须从字符向量中提取。我写这个：

add_var=function(index){
  label=c("products","price")
  var_name=label[index]
  df=data.frame(x=c(1,2,5,9),y=c(5,2,6,7))
  dt=as.data.table(df)
  dt[,(as.name(var_name)):=5]
  return(dt)
}
new_ds=add_var(1)

我期待类似的东西

x y products
1 5        5
2 2        5
5 6        5
9 7        5

但是，相反，我收到了这条错误消息：

 Error in `[.data.table`(dt, , `:=`((as.name(var_name)), 5)) : 
  LHS of := must be a symbol, or an atomic vector (column names or positions). 

有人知道如何修复我的功能以使其正常工作吗？

Answer 1

你只需要这个：

label <- c("products","price")
df <- data.frame(x=c(1,2,5,9),y=c(5,2,6,7))


setDT(df)[ , (label) := 5]


#>    x y label products price
#> 1: 1 5     5        5     5
#> 2: 2 2     5        5     5
#> 3: 5 6     5        5     5
#> 4: 9 7     5        5     5