在R中，如何限制data.table的单元格值的选择和设置到列的列表？

Question

I need to set all instances of "-1" in a subset of columns of a data.table to NA.

I can set all such instances of ALL columns in the data.table to NA as follows:

dt <- data.table(c("-1","A","A","B"), c("A","B","-1","-1"),c("-1","B","B","-1") )
for (i in seq_along(dt)) 
    set(dt, i=which(dt[[i]]=="-1"), j=i, value=NA)
> dt
   V1 V2 V3
1: NA  A NA
2:  A  B  B
3:  A NA  B
4:  B NA NA

But how do I limit the replacement to a subset of columns, eg c("V2","V3")? This doesn't work:

dt <- data.table(c("-1","A","A","B"), c("A","B","-1","-1"),c("-1","B","B","-1") )
for (i in seq_along(dt[,c("V2","V3"),with=FALSE])) 
    set(dt[,c("V2","V3"),with=FALSE], i=which(dt[,c("V2","V3"),with=FALSE][[i]]=="-1"), j=i, value=NA)
dt
> dt
V1 V2 V3
1: -1  A -1
2:  A  B  B
3:  A -1  B
4:  B -1 -1

Answer 1

We can loop through the index of names and set the elements that are -1 in those columns to NA.

for(j in paste0('V', 2:3)){
 set(dt, i=which(dt[[j]]==-1), j=j, value=NA)
}

dt
#   V1 V2 V3
#1: -1  A NA
#2:  A  B  B
#3:  A NA  B
#4:  B NA NA

EDIT: Modified based on @Frank's comments.

Answer 2

Here is a way without using a for loop:

dt <- data.table(c("-1","A","A","B"), c("A","B","-1","-1"), c("-1","B","B","-1") )

dt$V2 = ifelse(dt$V2 == "-1", NA, dt$V2)
dt$V3 = ifelse(dt$V3 == "-1", NA, dt$V3)

dt




   V1 V2 V3
1: -1  A NA
2:  A  B  B
3:  A NA  B
4:  B NA NA