在以下示例中:
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) 不会是一个很好的解决方案。
我们可以按 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)