Pilih baris matriks yang memenuhi suatu kondisi

Dalam R dengan sebuah matriks:

     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

Saya ingin mengekstrak submatrix yang barisnya memiliki kolom tiga = 11. Yaitu:

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

Saya ingin melakukan ini tanpa perulangan. Saya baru mengenal R jadi ini mungkin sangat jelas tetapi dokumentasi seringkali agak singkat.

Ini lebih mudah dilakukan jika Anda mengonversi matriks ke bingkai data menggunakan as.data.frame (). Jika demikian, jawaban sebelumnya (menggunakan subset atau m $ tiga) akan berfungsi, jika tidak maka tidak akan.

Untuk melakukan operasi pada sebuah matriks , Anda dapat mendefinisikan kolom dengan nama:

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

Atau dengan nomor:

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

Perhatikan bahwa jika hanya satu baris yang cocok, hasilnya adalah vektor integer, bukan matriks.

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

Perintah berikut akan memilih baris pertama dari matriks di atas.

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

Dan ini akan memilih tiga yang terakhir.

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

Hasilnya akan menjadi matriks dalam kedua kasus. Jika Anda ingin menggunakan nama kolom untuk memilih kolom maka Anda sebaiknya mengonversinya menjadi dataframe

mf <- data.frame(m)

Maka Anda dapat memilih dengan

mf[ mf$a == 16, ]

Atau, Anda bisa menggunakan perintah subset.

Saya akan memilih pendekatan sederhana menggunakan paket dplyr.

Jika bingkai data adalah data.

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

Subset adalah fungsi yang sangat lambat, dan saya pribadi merasa itu tidak berguna.

Saya menganggap Anda memiliki data.frame, array, matriks disebut Matdengan A, B, Csebagai nama kolom; maka yang perlu Anda lakukan adalah:

• Dalam kasus satu kondisi pada satu kolom, katakanlah kolom A

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

Dalam kasus beberapa kondisi pada kolom yang berbeda, Anda bisa membuat variabel dummy. Misalkan kondisi A = 10, B = 5dan C > 2, maka kita harus:

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

Dengan menguji keunggulan kecepatan system.time, whichmetode ini 10x lebih cepat dari subsetmetode.

Jika matriks Anda dipanggil m, cukup gunakan:

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

Jika dataset disebut data, maka semua baris yang memenuhi suatu kondisi di mana nilai kolom 'pm2.5'> 300 dapat diterima oleh -

data [data ['pm2.5']> 300,]