Sélectionnez les lignes d'une matrice qui remplissent une condition

En R avec une matrice:

     one two three four
 [1,]   1   6    11   16
 [2,]   2   7    12   17
 [3,]   3   8    11   18
 [4,]   4   9    11   19
 [5,]   5  10    15   20

Je veux extraire la sous-matrice dont les lignes ont la colonne trois = 11. C'est-à-dire:

      one two three four
 [1,]   1   6    11   16
 [3,]   3   8    11   18
 [4,]   4   9    11   19

Je veux faire cela sans boucler. Je suis nouveau sur R donc c'est probablement très évident, mais la documentation est souvent quelque peu laconique.

C'est plus facile à faire si vous convertissez votre matrice en un bloc de données en utilisant as.data.frame (). Dans ce cas, les réponses précédentes (utilisant un sous-ensemble ou m $ trois) fonctionneront, sinon elles ne le seront pas.

Pour effectuer l'opération sur une matrice , vous pouvez définir une colonne par son nom:

m[m[, "three"] == 11,]

Ou par numéro:

m[m[,3] == 11,]

Notez que si une seule ligne correspond, le résultat est un vecteur entier, pas une matrice.

m <- matrix(1:20, ncol = 4) 
colnames(m) <- letters[1:4]

La commande suivante sélectionnera la première ligne de la matrice ci-dessus.

subset(m, m[,4] == 16)

Et cela sélectionnera les trois derniers.

subset(m, m[,4] > 17)

Le résultat sera une matrice dans les deux cas. Si vous souhaitez utiliser des noms de colonnes pour sélectionner des colonnes, il vaut mieux la convertir en un dataframe avec

mf <- data.frame(m)

Ensuite, vous pouvez sélectionner avec

mf[ mf$a == 16, ]

Ou, vous pouvez utiliser la commande subset.

Je vais choisir une approche simple en utilisant le package dplyr.

Si le dataframe est data.

library(dplyr)
result <- filter(data, three == 11)

Le sous-ensemble est une fonction très lente, et personnellement je le trouve inutile.

Je suppose que vous avez un data.frame, tableau, appelée matrice Matavec A, B, Ccomme les noms de colonnes; alors tout ce que vous avez à faire est:

• Dans le cas d'une condition sur une colonne, disons la colonne A

  Mat[which(Mat[,'A'] == 10), ]

Dans le cas de plusieurs conditions sur une colonne différente, vous pouvez créer une variable factice. Supposons que les conditions soient A = 10, B = 5et C > 2, alors nous avons:

    aux = which(Mat[,'A'] == 10)
    aux = aux[which(Mat[aux,'B'] == 5)]
    aux = aux[which(Mat[aux,'C'] > 2)]
    Mat[aux, ]

En testant l'avantage de vitesse avec system.time, la whichméthode est 10 fois plus rapide que la subsetméthode.

Si votre matrice est appelée m, utilisez simplement:

R> m[m$three == 11, ]

Si l'ensemble de données est appelé data, alors toutes les lignes remplissant une condition où la valeur de la colonne 'pm2.5'> 300 peuvent être reçues par -

données [données ['pm2.5']> 300,]