R中基于跨表列函数计算为data.table新增布尔列及代码失效排查

原代码运行无报错但无法得到预期结果，核心存在两个本质问题：

• 自定义check函数不支持向量输入：函数内部使用的&&、||是R的标量短路逻辑运算符，仅会判断输入向量的第一个元素，不会逐行遍历整列计算。运行时因为运算符的短路截断特性，R不会抛出硬性报错，最终函数只会返回单个布尔值（示例中第一行参数为x1=-1、x2=-1，判断结果为FALSE），而非和列长度匹配的逐行结果。
• data.table的列赋值会自动将单值广播到整列长度，因此上述返回的单个FALSE会被填充到neg列的所有行，最终得到全为FALSE的错误结果，和预期的逐行布尔值不符。

第一步：改造自定义函数为向量化版本 #

将非向量化的标量逻辑运算符替换为逐元素计算的&、|，去掉冗余的if分支，让函数可以直接接收整列向量输入，返回同长度的布尔计算结果：

check <- function(x1, x2) {
  # 直接返回逐行比较的布尔向量，无需额外if判断
  (x1 < 0 & x2 > 0) | (x1 > 0 & x2 < 0)
}

第二步：跨表列对齐计算并新增列 #

在三个表的行序天然一一对应的前提下，直接取出需要计算的列传入函数，为目标表新增布尔列：

# 取DT1的第一列(a列)、DT2的第二列(b列)逐行计算，结果赋值给DT_RES的neg列
DT_RES[, neg := check(DT1[[1]], DT2[[2]])]

运行后输出和预期完全一致：

> DT_RES
#       id   neg
#    <int> <lgl>
# 1:     1 FALSE
# 2:     2  TRUE
# 3:     3  TRUE
# 4:     4 FALSE

注意：如果实际业务场景下三个表的行序不天然对齐，需要先通过公共关联键将DT1、DT2中需要计算的列合并到DT_RES中，再执行计算，不可直接按列位置取数，避免结果错位。

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