which函数的用法

关注公众号：数据科学专栏，回复"资料"，领取精美资料。

R语言的which函数主要是用于条件筛选，在R语言里面用的非常多，因为很多时候我们都需要筛选满足一些特定条件的数据记录。

 

向量

> a <- c(2,4,5,2,2,3)

> which(a == 2)

[1] 1 4 5

这里返回了向量a里面等于2的下标是1，4，5；还可以根据不等号筛选，比如：

> which(a>3)

[1] 2 3

这就筛选出了a里面大于3的数据的下标，怎么把这些数据取出呢？可以这样

> a[which(a>3)]

[1] 4 5

这样就取出了实际数据，当然了，有时候也需要取出下标！

 

矩阵

> mat <- matrix(c(1:24),nrow = 4,byrow = FALSE)

我们想取出第一列大于2的所有数据行，应该怎么做呢？