R:更新 data.table 中的列

Question

我有一个大数据表，结构如下

DT = data.table(Year=c("1993","1994"), "1"=c(NA,10), "2"=c(50, 40))

我想更新第二列“1”。每个带有“NA”的条目都应替换为“0”。但无论是

DT[is.na(1), 1:=0]

也不

DT[is.na("1"), "1":=0]

工作。问题是，列名 - 除了“Year” - 都是数字。当然，通过

setnames(DT, "1", "X1")
DT[is.na(X1), X1:=0]

这个小例子我可以解决这个问题，但是列名必须是数字，而且巨大的data.table有超过50列。有没有人有想法，我必须做什么？

Answer 1

你可以使用反引号

DT[is.na(`1`), `1`:=0]
DT
#   Year  1  2
#1: 1993  0 50
#2: 1994 10 40

如果有更多列，

nm1 <- names(DT1)[-1]
DT1[,(nm1):= lapply(.SD, function(x) replace(x, is.na(x), 0)), .SDcols=nm1]

DT1
#   Year  1  2  3 4
#1: 1993  0 50 10 0
#2: 1994 10 40  0 4

或者根据@Arun 的评论，多列的有效方法是使用set。与 replace 方法相比，它通过引用更新。

for(j in 2:ncol(DT1)){
  indx <- which(is.na(DT1[[j]]))
  set(DT1, i=indx, j=j, value=0)
}

数据

 DT1 <-  data.table(Year=c("1993","1994"), "1"=c(NA,10), 
             "2"=c(50, 40), "3"=c(10, NA), "4"=c(NA, 4))