Подмножество нескольких столбцов в R - более элегантный код?

Question

Я подмножество данных в соответствии с несколькими критериями в нескольких столбцах. Я выбираю строки в фрейме данных, которые содержат любое из нескольких значений, определенных в векторных «критериях» в любом из трех разных столбцов.

У меня есть код, который работает, но интересно, какие другие (более элегантные?) Способы сделать это. Вот что я сделал:

criteria <-c(1:10) 
subset1 <-subset(data, data[, "Col1"] %in% criteria | data[, "Col2"] 
%in% criteria | data[, "Col3"] %in% criteria) 

Предложения тепло приветствовали. (Я новичок R, поэтому очень просто объяснения о том, что вы предлагаете, также тепло приветствуются.)

Answer 1

Я не уверен, что если вам нужны две apply звонки здесь:

# Data 
df=data.frame(x=1:4,Col1=c(11,12,3,13),Col2=c(9,12,10,13),Col3=c(9,13,42,23)) 
criteria=1:10 

# Solution 
df[apply(df [c('Col1','Col2','Col3')],1,function(x) any(x %in% criteria)),] 

Если вы не хотите, чтобы сделать много столбцов, то это, вероятно, более удобным для чтения, чтобы сказать:

subset(df, Col1 %in% criteria | Col2 %in% criteria | Col3 %in% criteria) 

Answer 2

В качестве примера я использую DF, а не data.

DF[apply(apply(as.matrix(DF[c("Col1","Col2","Col3")]), 
       c(1,2), `%in%`, criteria), 
     1, any),] 

Для пробоя, что это делает:

Сделайте матрицу из указанных столбцов, и для каждого элемента в этом тесте матрицы, если она содержит один из критериев. Затем для каждой строки этой матрицы смотрите, есть ли какой-либо из элементов строки: TRUE. Если да, сохраните соответствующую строку исходного набора данных.

Работа на примере:

Начать с фиктивными данными:

DF <- data.frame(Col1=seq(1, by=2, length=10), 
       Col2=seq(3, by=3, length=10), 
       Col3=seq(7, by=1, length=10), 
       other=LETTERS[1:10]) 

который выглядит как

> DF 
    Col1 Col2 Col3 other 
1  1 3 7  A 
2  3 6 8  B 
3  5 9 9  C 
4  7 12 10  D 
5  9 15 11  E 
6 11 18 12  F 
7 13 21 13  G 
8 15 24 14  H 
9 17 27 15  I 
10 19 30 16  J 

Вытащите только столбцы, представляющие интерес.

> as.matrix(DF[c("Col1","Col2","Col3")]) 
     Col1 Col2 Col3 
[1,] 1 3 7 
[2,] 3 6 8 
[3,] 5 9 9 
[4,] 7 12 10 
[5,] 9 15 11 
[6,] 11 18 12 
[7,] 13 21 13 
[8,] 15 24 14 
[9,] 17 27 15 
[10,] 19 30 16 

Проверить каждый элемент по сравнению с критериями

> apply(as.matrix(DF[c("Col1","Col2","Col3")]), c(1,2), `%in%`, criteria) 
     Col1 Col2 Col3 
[1,] TRUE TRUE TRUE 
[2,] TRUE TRUE TRUE 
[3,] TRUE TRUE TRUE 
[4,] TRUE FALSE TRUE 
[5,] TRUE FALSE FALSE 
[6,] FALSE FALSE FALSE 
[7,] FALSE FALSE FALSE 
[8,] FALSE FALSE FALSE 
[9,] FALSE FALSE FALSE 
[10,] FALSE FALSE FALSE 

испытаний, если какой-либо из значений в строке истинны

> apply(apply(as.matrix(DF[c("Col1","Col2","Col3")]), c(1,2), `%in%`, criteria), 1, any) 
[1] TRUE TRUE TRUE TRUE TRUE FALSE FALSE FALSE FALSE FALSE 

использовать, чтобы индекс исходного кадра данных.

> DF[apply(apply(as.matrix(DF[c("Col1","Col2","Col3")]), c(1,2), `%in%`, criteria), 1, any),] 
    Col1 Col2 Col3 other 
1 1 3 7  A 
2 3 6 8  B 
3 5 9 9  C 
4 7 12 10  D 
5 9 15 11  E