基于data.table实现同ID组内ALPHA/BETA存在TRUE则AB为TRUE

原始数据 #

DT <- data.table::data.table(
          ID = c(1L, 1L, 1L, 2L, 2L, 2L),
       ALPHA = c(TRUE, FALSE, NA, NA, TRUE, FALSE),
        BETA = c(FALSE, NA, TRUE, NA, FALSE, NA)
)

需求说明 #

需要新增一列AB，规则为：同一ID分组中，只要ALPHA或BETA任意一列存在TRUE值，该分组所有行的AB都设为TRUE。

解决方案 #

直接用data.table的分组赋值语法即可实现：

DT[, AB := any(ALPHA | BETA, na.rm = TRUE), by = ID]

代码解释 #

• by = ID：指定按ID字段分组处理数据
• ALPHA | BETA：判断单条数据中ALPHA或BETA是否为TRUE，NA参与逻辑或运算时会返回NA，但后续的any函数会处理这种情况
• any(..., na.rm = TRUE)：对分组内的逻辑判断结果做“存在性”验证，na.rm = TRUE会忽略NA值，只要分组里有至少一个TRUE，整个分组的AB就会被赋值为TRUE

运行上述代码后，DT的结果与期望输出完全一致：

ID ALPHA BETA AB
1:  1  TRUE FALSE TRUE
2:  1 FALSE    NA TRUE
3:  1    NA  TRUE TRUE
4:  2    NA    NA TRUE
5:  2  TRUE FALSE TRUE
6:  2 FALSE    NA TRUE

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