有没有办法为按行以编程方式选择的 data.table 列分配值？ [复制]

Question

我正在处理一个包含大约 500 列和数千行的数据表，每列代表一个可能会或可能不会出现在字符串中的项目。例如，我的数据看起来像这样

   String                Item 1 Item 2  Item 3 Item 4
1  "Item 1Item 2Item 4"    0     0        0     0
2  "Item 1Item 2"          0     0        0     0      
3  "Item 3"                0     0        0     0      

我已将字符串解析为项目编号，结果项目位于列表中（因此，对于上面的第一个观察，列表项目 1 将包含元素“项目 1”、“项目 2”和“项目 4”。

我试图以编程方式更改每列的值，方法是将列表中的每一项用作行的列名，然后将 1 分配给这些列。例如，我可以构建一个简单的 for 循环来执行我正在寻找的操作：

for (i in 1:nrow(data)){
   data[i, eval(unlist(listofitems[[i]])) := 1]
}

返回

  String                Item 1 Item 2  Item 3 Item 4
1  "Item 1Item 2Item 4"    1     1        0     1
2  "Item 1Item 2"          1     1        0     0      
3  "Item 3"                0     0        1     0           

但是，考虑到数据的大小以及出现类似情况的频率（有时我希望能够对数据表对通过引用分配的列进行逐行操作），我希望可能会有一种更“data.table-y”的方式来获得最终答案。

Answer 1

我们可以使用cSplit_e

library(splitstackshape)
out <- cSplit_e(data[1], 'String', type = 'character', sep=":", fill = 0)
names(out)[-1] <- sub("String_", "", names(out)[-1])
out
#              String Item 1 Item 3 Item2 Item4
#1 Item 1:Item2:Item4      1      0     1     1
#2       Item 1:Item2      1      0     1     0
#3             Item 3      0      1     0     0

数据

data <- structure(list(String = c("Item 1:Item2:Item4", "Item 1:Item2", 
"Item 3"), Item1 = c(0L, 0L, 0L), Item2 = c(0L, 0L, 0L), Item3 = c(0L, 
0L, 0L), Item4 = c(0L, 0L, 0L)), class = "data.frame", row.names = c("1", 
"2", "3"))

Answer 2

一个选项是使用矩阵数字索引进行分配：

cols <- setdiff(names(DT), c("String", "ParsedString"))
DT[, (cols) := {
    m <- cbind(rep(1L:.N, lengths(ParsedString)), 
        match(unlist(ParsedString), names(.SD)))
    ans <- as.matrix(.SD)
    ans[m] <- 1L
    as.data.table(ans)
}, .SDcols=cols]

数据：

library(data.table)
DT <- fread('String,Item 1,Item 2,Item 3,Item 4
"Item 1:Item 2:Item 4",0,0,0,0
"Item 1:Item 2",0,0,0,0      
"Item 3",0,0,0,0')
DT[, ParsedString := strsplit(String, split=":")]