在多列上过滤具有相同条件的行

Question

我想在不同的列中使用相同的条件来选择行。

set.seed(123)
df <- data.frame(col.x = sample(LETTERS[1:10], 20, replace = TRUE), 
                 col.y = sample(LETTERS[1:10], 20, replace = TRUE), 
                 val = rnorm(20))

我只需要在ValsToRetain 两列col.x、col.y 中包含值。

ValsToRetain <- c('A','D', 'F','H','J')

我已经尝试了这两种方法，它们给出了相同的预期输出。

df %>% filter(col.x %in% ValsToRetain) %>% filter(col.y %in% ValsToRetain)
df %>% filter(col.x %in% ValsToRetain & col.y %in% ValsToRetain)
#   col.x col.y        val
# 1     A     H -1.6866933
# 2     F     F  0.8377870
# 3     J     J  0.4264642
# 4     F     H  0.8781335
# 5     D     D -0.3059627

但是，还有其他优雅的方法可以做到这一点吗？

例如；比如在这些列中计算rowSums 以检查它是否有nas。由于有多个值，我无法得到类似rowSums(df[,1:2] == 'A') 的想法。

Answer 1

我们可以使用filter_at 和all_vars

df %>% 
  filter_at(vars(starts_with("col")), all_vars(. %in% ValsToRetain))
#   col.x col.y        val
#1     A     H -1.6866933
#2     F     F  0.8377870
#3     J     J  0.4264642
#4     F     H  0.8781335
#5     D     D -0.3059627

Answer 2

这是使用Reduce 和lapply 的基本R 方法。 lapply 在相关列中应用 %in%，返回逻辑向量列表。然后Reduce 使用& 将这些向量组合成一个向量。

df[Reduce("&", lapply(df[c("col.x", "col.y")], "%in%", ValsToRetain)),]
   col.x col.y        val
6      A     H -1.6866933
7      F     F  0.8377870
11     J     J  0.4264642
14     F     H  0.8781335
19     D     D -0.3059627

如果您有很多列进行比较，那么您可以使用grep 来选择它们，例如grep("^col", names(df)) 代替c("col.x", "col.y")。

Answer 3

您可以使用子集，它很短但与您的方法相同：

filt <- subset(df,col.x %in% ValsToRetain & col.y %in% ValsToRetain)

输出是：

 col.x col.y        val
     A     H -1.6866933
     F     F  0.8377870
     J     J  0.4264642
     F     H  0.8781335
     D     D -0.3059627