根据data.table中一列的值更改多列的值

Question

假设我有一个数据表，dt1:

dt1 <- data.table(
names = c("A1", "XX", "A2", "XY", "A3", "XZ"),
   A1 = c( 0,    0,    0,    0,    0,    0), 
   A2 = c( 0,    0,    0,    0,    0,    0), 
   A3 = c( 0,    0,    0,    0,    0,    0)
)

我想要这样的新数据表：

dt2 <- data.table(
names = c("A1", "XX", "A2", "XY", "A3", "XZ"),
   A1 = c( 1,    0,    0,    0,    0,    0), 
   A2 = c( 0,    0,    1,    0,    0,    0), 
   A3 = c( 0,    0,    0,    0,    1,    0)
)

即如果names列的行值与某个列的名称相同，则该列的行值更改为1。

我可以通过以下代码实现：

dt1[names == "A1", "A1" := 1]
dt1[names == "A2", "A2" := 1]
dt1[names == "A3", "A3" := 1]

但我想知道是否有更简单的方法可以做到这一点，尤其是当我想要更改的列数很大时。

我尝试了以下几行，但它们不起作用：

cln <- c("A1", "A2", "A3")
dt1[names == (cln), (cln) := 1]

Answer 1

使用data.table的高效for(...) set(...)组合：

for(j in names(dt1)[-1]) {
  set(dt1, dt1[, .I[names == j]], j, value = 1)
}

给出：

> dt1
   names A1 A2 A3
1:    A1  1  0  0
2:    XX  0  0  0
3:    A2  0  1  0
4:    XY  0  0  0
5:    A3  0  0  1
6:    XZ  0  0  0

您也可以使用setdiff(names(dt1), "names")，而不是names(dt1)[-1]。

Answer 2

你可以用循环来做到这一点。

for(i in colnames(dt1)[-1]) {
 dt1[,i] <- ifelse(dt1[,"names"] == i, 1, 0)
}