Sous-ensemble de plusieurs colonnes dans la R - plus élégant de code?

Je suis. un dataframe en fonction de multiples critères sur plusieurs colonnes. J'ai choisi les lignes dans le dataframe qui contiennent l'une des valeurs définies dans le vecteur "critères", l'une des trois colonnes différentes.

J'ai un code qui fonctionne, mais me demande ce que les autres (plus élégant?) moyens qui existent pour ce faire. Voici ce que j'ai fait:

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

Suggestions chaleureusement accueillis. (Je suis un R débutant, donc très simples, des explications sur ce que vous suggérez sont également accueilli chaleureusement.)

OriginalL'auteur user1257313 | 2012-03-09

  1. 13

    Je ne suis pas sûr si vous avez besoin de deux apply appelle ici:

    # 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)),]

    Sauf si vous voulez faire un grand nombre de colonnes, alors il est probablement plus lisible-à-dire:

    subset(df, Col1 %in% criteria | Col2 %in% criteria | Col3 %in% criteria)
    notez que la page de man pour subset met en garde contre son utilisation par programmation, car il utilise non standard d'évaluation
    J'ai lu que la prudence sur la page d'aide, mais je ne sais pas ce que cela signifie. Pourquoi "non-standard d'évaluation" être un problème? Dans mon cas, je suis juste de sectionnement de données pour créer des statistiques descriptives, donc je ne pense pas "sous-ensemble", sera à l'origine de problèmes pour moi... mais dans ce cas, serait-il poser des problèmes? Merci.

    OriginalL'auteur nograpes

  2. 6

    Je suis en utilisant DF plutôt que data que dans l'exemple.

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

    Pour une ventilation de ce qui ce fait:

    Faire une matrice de colonnes spécifiées, et pour chaque élément de la matrice de test si elle contient l'un des critères. Ensuite, pour chaque ligne de cette matrice, voir si l'un des éléments de ligne sont TRUE. Si oui, garder la ligne correspondante de l'ensemble de données initial.

    De travail par le biais d'un exemple:

    Commencer avec les données factices:

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

    qui ressemble

    > 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

    Sortir juste les colonnes d'intérêt.

    > 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

    Vérifier chaque entrée contre les critères

    > 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

    Tester si une valeur dans une ligne sont de VÉRITABLES

    > 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

    L'utiliser pour indexer les données d'origine image.

    > 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

    OriginalL'auteur Brian Diggs