r - OR 在数据框选择表达式中

Question

DF1 <- DF[DF$cat == 'A', ]
DF2 <- DF[DF$cat == 'B', ]
RDF <- rbind(DF1, DF2)

有没有办法以更直接的方式表达这一点，例如

RDF <- DF[DF$ cat == c('A','B'), ] # Does not work

Answer 1

RDF <- DF[DF$cat %in% c('A','B'), ]

Answer 2

更一般地，对于两个条件：

RDF <- DF[ DF$cat == "A" | DF$dog == "B", ]

Answer 3

您可以使用 data.table 和键

library(data.table)
DT <- data.table(DF)
setkey(DT, cat)

DT[c("A", "B"),]