r中的多个条件

Question

有没有更好的方法来拥有多个条件以简化我的代码。 这是我所说的一个例子。

library(data.table)
test <- data.table(dx1 <- c("a", "b", "c"),
                   dx2 <- c("b", "c", "d"),
                   dx3 <- c("a", "f", "g"))

condition <- c("a", "b")

# This is what I want to simplify
test[,dx4 := ifelse(dx1 %in% condition|
                    dx2 %in% condition |
                    dx3 %in% condition, 1, 0)]

谢谢

Answer 1

我们可以使用Reduce 和|。在.SDcols 中指定感兴趣的列（如果使用了所有列，则不需要），遍历Data.table 的子集（.SD），检查“条件”元素是否为%in% 列，@ 987654326@ 使用| 将其转换为逻辑vector 并使用as.integer 或+ 转换为二进制

test[, dx4 := +(Reduce(`|`, lapply(.SD, `%in%`, condition))), .SDcols = dx1:dx3]
test
#   dx1 dx2 dx3 dx4
#1:   a   b   a   1
#2:   b   c   f   1
#3:   c   d   g   0

数据

test <- data.table(dx1 = c("a", "b", "c"),
               dx2 = c("b", "c", "d"),
               dx3 = c("a", "f", "g"))