複数の列で同じ条件の行をフィルタリングする

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)) 

は、私は2つの列col.x、col.y渡っValsToRetainで値だけを持っている必要があります。

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

私は、同じ期待される出力を与えるこれら2つのアプローチを試しました。

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を計算して、naがあるかどうかを確認します。複数の価値があるので、私はrowSums(df[,1:2] == 'A')のような考えを得ることができませんでした。

Answer 1

私たちは、あなたがサブセットを使用することができます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

でfilter_atを使用することができ、それはかなり短いですが、あなたの方法と同じ方法が：

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 

Answer 3

Reduceとを使った基底R法です。 lapplyは、関連する列にまたがって%in%を適用し、論理ベクトルのリストを返します。 Reduceは、&を使用してこれらのベクターを1つのベクターに結合します。

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 

あなたは比較のために多くの列を持っている場合は、c("col.x", "col.y")の代わりにgrep("^col", names(df))のようにそれらを選択するために、grepを使用することができます。