R data.table如何使用ifelse按引用同时赋值两个列？

错误原因 #

你代码报错的核心原因是基础ifelse函数的返回值结构不符合多列赋值要求：

• ifelse是向量化函数，返回结果的长度和条件向量（这里是a > 2，共6个元素，对应数据表6行）长度完全一致
• 你传入的yes/no分支是长度为2的列表，ifelse会自动把列表循环到和条件向量等长，最终返回长度为6的一维结果，无法匹配你要赋值的2列的结构，因此抛出长度不匹配的错误。

解决方法 #

方法1：代码块内复用条件，分别定义两列返回值 #

在:=右侧用大括号包裹逻辑，先定义一次判断条件，再用data.table的列表语法.(列1, 列2)返回两列结果，既避免重复写判断逻辑，也符合多列赋值的结构要求：

library(data.table)
x <- data.table(a = 1:3, b = 1:6) 

x[, c("col1", "col2") := {
  cond <- a > 2
  .(
    fifelse(cond, "Yes", "No"),
    fifelse(cond, b, a)
  )
}]

这里用data.table原生的fifelse替换基础ifelse，运行效率更高，也能避免基础ifelse的类型自动转换问题。

方法2：分两次按条件赋值 #

如果偏好更直观的写法，可以先给所有行赋默认值，再单独更新满足条件的行，这种写法在数据量较大时效率更高：

x <- data.table(a = 1:3, b = 1:6) 

# 先赋值不满足a>2的默认值
x[, c("col1", "col2") := .("No", a)]
# 仅更新满足a>2的行
x[a > 2, c("col1", "col2") := .("Yes", b)]

运行结果验证 #

两种方法运行后得到的结果完全一致：

> x
   a b col1 col2
1: 1 1   No    1
2: 2 2   No    2
3: 3 3  Yes    3
4: 1 4   No    1
5: 2 5   No    2
6: 3 6  Yes    6

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