选择满足条件的矩阵行

Question

在带有矩阵的 R 中：

     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

我想提取其行具有第三列 = 11 的子矩阵。即：

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

我想在不循环的情况下做到这一点。 我是 R 的新手，所以这可能非常明显，但文档通常有些简洁。

Answer 1

如果您使用 as.data.frame() 将矩阵转换为数据框，这将更容易做到。 在这种情况下，先前的答案（使用子集或 m$three）将起作用，否则将不起作用。

要对矩阵执行操作，您可以按名称定义列：

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

或按编号：

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

请注意，如果只有一行匹配，则结果是整数向量，而不是矩阵。

Answer 2

我将选择使用 dplyr 包的简单方法。

如果数据帧是数据。

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

Answer 3

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

以下命令将选择上面矩阵的第一行。

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

这将选择最后三个。

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

在这两种情况下，结果都将是一个矩阵。 如果您想使用列名来选择列，那么最好将其转换为数据框

mf <- data.frame(m)

然后你可以选择

mf[ mf$a == 16, ]

或者，您可以使用子集命令。

Answer 4

Subset 是一个非常慢的函数，我个人觉得它没用。

我假设您有一个名为Mat的 data.frame、array、矩阵，其中A 、 B 、 C作为列名； 那么你需要做的就是：

• 在一列上有一个条件的情况下，假设列 A

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

在不同列有多个条件的情况下，您可以创建一个虚拟变量。 假设条件是A = 10 ， B = 5 ，和C > 2 ，那么我们有：

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

通过使用system.time测试速度优势， which方法比subset方法快 10 倍。

Answer 5

如果您的矩阵称为m ，只需使用：

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

Answer 6

如果数据集被称为数据，那么所有满足列 'pm2.5' > 300 的值的行都可以通过 -

数据[数据['pm2.5']>300,]