R 如何比较一组列与另一组列

发布于09月15日

假设我们有以下数据

df <- data.frame(jan=c(10,7,1),
           feb=c(20,9,2),
           mar=c(30,6,3),
           apr=c(5,5,4),
           may=c(0,9,5),
           jun=c(10,8,6)
)


  jan feb mar apr may jun
1  10  20  30   5   0  10
2   7   9   6   5   9   8
3   1   2   3   4   5   6

我期望得到一个新的标志变量，该变量是通过将1组列(1月2月3月)与(4月5月6月)进行比较而得出的，如果1组列的值与另一组列的值匹配，则FLAG=‘Y’，否则‘N’.

在这里，Jan值与第一行的Jun匹配，因此标志为Y.

  jan feb mar apr may jun flag
1  10  20  30   5   0  10    Y
2   7   9   6   5   9   8    Y
3   1   2   3   4   5   6    N

推荐答案

首先，我们遍历数据帧的各行，并在每次迭代中将数据帧unlist转换 for each "集合"的向量.然后使用%in%运算符查看第一个集合中的any个元素是否出现在第二个集合中.最后，将结果赋给df的第flag列.

ifelse部分仅用于将逻辑输出转换为"Y"和"N".如果您对逻辑输出满意，则可以忽略它.

df$flag <- sapply(1:nrow(df), \(x) 
                  ifelse(any(unlist(df[x, 1:3]) %in% unlist(df[x, 4:6])), "Y", "N"))
df
#>   jan feb mar apr may jun flag
#> 1  10  20  30   5   0  10    Y
#> 2   7   9   6   5   9   8    Y
#> 3   1   2   3   4   5   6    N