Chọn các hàng của ma trận đáp ứng điều kiện

Trong R với một ma trận:

     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

Tôi muốn trích xuất hàm con có các hàng có cột ba = 11. Đó là:

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

Tôi muốn làm điều này mà không cần lặp. Tôi mới sử dụng R nên điều này có lẽ rất rõ ràng nhưng tài liệu thường có phần ngắn gọn.

Điều này dễ thực hiện hơn nếu bạn chuyển đổi ma trận của mình sang khung dữ liệu bằng as.data.frame (). Trong trường hợp đó, các câu trả lời trước (sử dụng tập hợp con hoặc m $ ba) sẽ hoạt động, nếu không thì không.

Để thực hiện thao tác trên ma trận , bạn có thể xác định một cột theo tên:

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

Hoặc theo số:

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

Lưu ý rằng nếu chỉ có một hàng khớp, kết quả là một vectơ số nguyên, không phải là ma trận.

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

Lệnh sau sẽ chọn hàng đầu tiên của ma trận ở trên.

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

Và điều này sẽ chọn ba cuối cùng.

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

Kết quả sẽ là một ma trận trong cả hai trường hợp. Nếu bạn muốn sử dụng tên cột để chọn cột thì tốt nhất bạn nên chuyển đổi nó thành một khung dữ liệu với

mf <- data.frame(m)

Sau đó, bạn có thể chọn với

mf[ mf$a == 16, ]

Hoặc, bạn có thể sử dụng lệnh tập hợp con.

Tôi sẽ chọn một cách tiếp cận đơn giản bằng cách sử dụng gói dplyr.

Nếu khung dữ liệu là dữ liệu.

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

Tập hợp con là một chức năng rất chậm, và cá nhân tôi thấy nó vô dụng.

Tôi giả sử bạn có một data.frame, mảng, ma trận được gọi Matvới A, B, Cnhư tên cột; sau đó, tất cả những gì bạn cần làm là:

• Trong trường hợp có một điều kiện trên một cột, giả sử cột A

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

Trong trường hợp có nhiều điều kiện trên cột khác nhau, bạn có thể tạo một biến giả. Giả sử các điều kiện là A = 10, B = 5và C > 2, sau đó chúng ta có:

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

Bằng cách kiểm tra lợi thế tốc độ với system.time, whichphương pháp nhanh hơn gấp 10 lần so với subsetphương pháp.

Nếu ma trận của bạn được gọi m, chỉ cần sử dụng:

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

Nếu tập dữ liệu được gọi là dữ liệu, thì tất cả các hàng đáp ứng điều kiện trong đó giá trị của cột 'pm2,5'> 300 có thể được nhận bởi -

dữ liệu [dữ liệu ['pm2,5]]> 300,]