En R, seleccione filas de una matriz que cumplen una condición

Question

en R con una matriz:

 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 

I desea extraer la submatriz cuyas filas tienen la columna tres = 11. Esto es:

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

Quiero hacer esto sin bucles. Soy nuevo en R, así que esto es probablemente muy obvio, pero la documentación de suele ser algo escueta.

Answer 1

Esto es fácil de hacer si usted convierte su matriz a una trama de datos utilizando as.data.frame(). En ese caso, las respuestas anteriores (con subconjunto o m $ 3) funcionarán, de lo contrario no lo harán.

para realizar la operación en una matriz, puede definir una columna por su nombre:

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

O por número:

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

Tenga en cuenta que si sólo una fila coincide, el resultado es un vector entero, no una matriz.

Answer 2

Si su matriz se llama m, sólo tiene que utilizar:

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

Answer 3

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

El siguiente comando seleccionará la primera fila de la matriz de arriba.

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

Y esto seleccionará los tres últimos.

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

El resultado será una matriz en ambos casos. Si desea utilizar nombres de columna para seleccionar columnas, entonces sería mejor fuera de su conversión en una trama de datos con

mf <- data.frame(m) 

continuación, puede seleccionar con

mf[ mf$a == 16, ] 

O bien, puede utilizar el comando subconjunto .

Answer 4

El subconjunto es una función muy lenta, y personalmente me parece inútil.

que suponga que tiene un hoja.de.datos, matriz, matriz de llamada Mat con A, B, C como nombres de columna; entonces todo lo que necesita hacer es:

• En el caso de una condición en una columna, digamos que la columna A

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

En el caso de múltiples condiciones en diferentes columna, puede crear una variable ficticiaSupongamos que las condiciones son A = 10, B = 5 y C > 2, entonces tenemos:

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

Al poner a prueba la ventaja de la velocidad con system.time, el método which es 10 veces más rápido que el método subset.

Answer 5

Elegiré un enfoque simple usando el paquete dplyr.

Si el marco de datos son datos.

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