R：基于同一列的多个条件的子集

Question

我正在尝试过滤如下所示的数据框：

KeyID   SQ Value.S  Course  Type    Survey.Num      
10001   A1  1      1024    Classic    1     
10002   A1  1      1024    Classic    1     
10002   A1  1      1024    Classic    2     
10002   A1  1      1024    Classic    3     
10005   A1  1      1024    Classic    1     
10005   A1  1      1024    Classic    2     

我只希望KeyID在Survey.Num 1、2和3中具有值的行。

所以我的输出看起来像这样：

KeyID   SQ  Value.S Course  Type    Survey.Num  

10002   A1    1    1024    Classic  1    
10002   A1    1    1024    Classic  2
10002   A1    1    1024    Classic  3    

我尝试使用filter（）函数，但似乎不起作用。

有什么想法吗？

Answer 1

或数据data.table解决方案

library(data.table)
setDT(df)[, .SD[all(seq_len(3) %in% Survey.Num)], by = KeyID]

##    KeyID SQ Value.S Course    Type Survey.Num
## 1: 10002 A1       1   1024 Classic          1
## 2: 10002 A1       1   1024 Classic          2
## 3: 10002 A1       1   1024 Classic          3

Answer 2

一种选择是使用ave （假设您的data.frame称为df ）：

df[as.logical(ave(df$Survey.Num, df$KeyID, FUN = function(x) all(1:3 %in% x))),]
#  KeyID SQ Value.S Course    Type Survey.Num
#2 10002 A1       1   1024 Classic          1
#3 10002 A1       1   1024 Classic          2
#4 10002 A1       1   1024 Classic          3

或使用dplyr ：

df %>% group_by(KeyID) %>% filter(all(1:3 %in% Survey.Num))
#Source: local data frame [3 x 6]
#Groups: KeyID
#
#  KeyID SQ Value.S Course    Type Survey.Num
#1 10002 A1       1   1024 Classic          1
#2 10002 A1       1   1024 Classic          2
#3 10002 A1       1   1024 Classic          3