比较 R data.table 中多列值的向量

Roy*_*alk 2 r data.table

在以下示例中:

DT <- data.table(A1=c(1,2),A2=c(2,3),B1=c(1,2),B2=c(3,4),B3=c(2,5))
Run Code Online (Sandbox Code Playgroud)

我想检查每一行 A1 和 A2 列中的值是否都出现在 B1、B2 或 B3 中。

如果我做:

DT[,BOTH_PRESENT:=!(F %in% (c(A1,A2) %in% c(B1,B2,B3)))]
Run Code Online (Sandbox Code Playgroud)

这产生:

   A1 A2 B1 B2 B3 BOTH_PRESENT
1:  1  2  1  3  2         TRUE
2:  2  3  2  4  5         TRUE
Run Code Online (Sandbox Code Playgroud)

虽然我希望第一行为 True,第二行为 False。

# Row 1
!(F %in% (c(1,2) %in% c(1,3,2)))
# Row 2
!(F %in% (c(2,3) %in% c(2,4,5)))
Run Code Online (Sandbox Code Playgroud)

我应该如何更改我的代码?

由于我的真实数据集在列方面要大得多,因此像这样的表达式(A1 == B1 | A1 == B2 | A1 == B3) & (A2 == B1 | A2 == B2 | A2 == B3) 不会是一个很好的解决方案。

akr*_*run 5

我们可以按 data.table 的行序列、unlist“A”列进行分组,检查all这些元素是否是%in%ed unlist“B”列。

 DT[, BOTH_PRESENT := all(unlist(.SD[, 1:2, with = FALSE]) %in% 
        unlist(.SD[, 3:5, with = FALSE])), by = 1:nrow(DT)]
Run Code Online (Sandbox Code Playgroud)

或者正如@Roland 提到的,

 DT[,BOTH_PRESENT:= all(c(A1, A2) %in% c(B1,B2,B3)), by = seq_len(nrow(DT))]
Run Code Online (Sandbox Code Playgroud)

  • 或者 `DT[,BOTH_PRESENT:= all(c(A1, A2) %in% c(B1,B2,B3)), by = seq_len(nrow(DT))]`。机器人注意到这是一个循环迭代行,这可能会很慢。 (3认同)