R语言中替换data.table单元格值：将1替换为列名、0替换为NA的实现问题

先看你的初始数据和需求：

你初始化的data.table代码是：

set.seed(45)
DT = data.table(
 names = c("n1", "n2", "n3", "n4", "n5"),
 a = sample(c(0, 1), size = 5, replace = TRUE),
 b = sample(c(0, 1), size = 5, replace = TRUE),
 c = sample(c(0, 1), size = 5, replace = TRUE)
)

初始输出：

names a b c
1: n1 0 1 1
2: n2 0 0 0
3: n3 1 0 0
4: n4 1 1 1
5: n5 0 0 1

期望把每列的1替换为列名，0替换为NA，得到：

names a b c
1: n1 NA b c
2: n2 NA NA NA
3: n3 a NA NA
4: n4 a b c
5: n5 NA NA c

你的代码错误原因分析 #

你尝试用循环逐行逐列处理时遇到了两个核心问题：

• data.table赋值语法错误：你用了DT[r, c] <- ...这种方式，这不是data.table推荐的原地修改原表的方法，正确的原地修改应该用:=操作符。直接用<-会创建临时副本，还容易因为类型不匹配（整数列赋值字符串）导致数据异常，这就是你取消注释后出现NA的原因。
• 循环效率低下且易出错：逐行逐列的循环在data.table里是反模式，data.table设计初衷就是为了矢量化操作，循环不仅速度慢，还容易出现索引错误。

正确的解决方案 #

推荐两种高效的方法，都不需要逐行循环：

方法1：用lapply批量处理目标列

我们可以针对需要修改的列（这里是a、b、c），用lapply遍历，结合ifelse完成替换：

# 获取需要修改的列名（排除第一列names）
target_cols <- setdiff(colnames(DT), "names")

# 批量修改每一列
DT[, (target_cols) := lapply(.SD, function(x) {
  col_name <- deparse(substitute(x))
  ifelse(x == 1, col_name, NA_character_)
}), .SDcols = target_cols]

方法2：更简洁的向量化替换

如果列名和变量名一致，还可以用更直接的方式，遍历列名并赋值：

for (col in target_cols) {
  DT[, (col) := ifelse(get(col) == 1, col, NA_character_)]
}

这里用get(col)获取列的值，(col) :=确保列名被正确解析为变量，NA_character_明确指定字符型的NA，避免类型转换问题。

验证结果 #

运行上面的代码后，你就能得到期望的输出：

names    a    b    c
1:    n1 <NA>    b    c
2:    n2 <NA> <NA> <NA>
3:    n3    a <NA> <NA>
4:    n4    a    b    c
5:    n5 <NA> <NA>    c

内容的提问来源于stack exchange，提问作者Edlenil