data.table基于"by"参数创建列名

Question

假设我有以下数据

dt <- data.table(var = c("a", "b"), val = c(1, 2))

现在我想向dt添加两个新列，分别命名为a和b，具有各自的值（1,2）。我可以使用循环来执行此操作，但我想以data.table的方式执行。

结果将是这样的data.table：

dt.res <- data.table(var = c("a", "b"), val = c(1, 2), #old vars 
                     a = c(1, NA), b = c(NA, 2)) # newly created vars

到目前为止，我想出了类似这样的东西

dt[, c(xx) := val, by = var]

其中< code>xx是一个< code > data . table -类似于< code >的命令。N，它寻址< code > by -组的值。

谢谢你的帮助！

附录：for-loop 方式

具有< code > for -循环而不是< code > by -参数的非data.table-way看起来像这样:

for (varname in dt$var){
     dt[var == varname, c(varname) := val]
}

Answer 1

基于所展示的示例，我们可以使用data.table中的dcast将长格式转换为宽格式，并在'val'列上与原始数据集连接。

library(data.table)#v1.9.6+
dt[dcast(dt, val~var, value.var='val'), on='val']
#   var val  a  b
#1:   a   1  1 NA
#2:   b   2 NA  2

或者就像在注释中提到的那样，对于以前的版本，要么合并，要么设置键列，然后加入。

merge(dt, dcast.data.table(dt, val~var, value.var='val'))