Комплексное подмножество данных

Question

Рассмотрим следующий кадр данных, где у меня есть 7 переменных id, от A до F

id     A           B          C          D         E         F          
1    5590       23658      523        727       52903     732569
2    24311      421        4319       5597      32695     4521
3    626         623       78

И я хотел бы новую переменную G, где G включает только наблюдения, которые имеют значение 5590 или 421 в A через F.

Так что G будет включать в себя только наблюдения с id = 1 и 2 в этом случае.

Есть ли быстрый способ сканирования переменных от A до F на R?

Answer 1

dfr <- read.table(text= "5590       23658      523        727       52903     732569
24311      421        4319       5597      32695     4521
24311      431        4319       5597      32695     4521
24311      4211        431239       5597      32695     43521")

dfr <- as.numeric(dfr[,1:6])
#in case it exists remove it.
remove('subsetdfr')

i <- 0

#dim(dfr)[2] automatically output the number of columns of the dataframe
while (i < dim(dfr)[2]){
  i <- i+1

  if (exists('subsetdfr') == TRUE ) {
    #add a row in case subsetdfr exists
    subsetdfr <- rbind(subsetdfr,subset(dfr, dfr[i] == 421 | dfr[i] == 5590 )) 
    } else {
    #create subsetdfr in case it does not exsits
    subsetdfr <- data.frame(subset(dfr, dfr[i] == 421 | dfr[i] == 5590 ))
    }

}

subsetdfr

этот результат в:

> subsetdfr
     V1    V2   V3   V4    V5     V6
1  5590 23658  523  727 52903 732569
2 24311   421 4319 5597 32695   4521

Answer 2

Мы можем использовать apply

df1$G  <- apply(df1[-1], 1, function(x) intersect(x, c(5590, 421))[1])