基于一列的条件对数据帧进行子集化

时间:2016-06-25 15:24:56

标签: r dataframe subset apply

我有一个很大的数据框架。我想根据一列值的条件制作一个子集。说:

a<-data.frame(x=rep(1:5,5),y=rnorm(25),z=runif(25))

我想根据列x的值制作一个子集。例如,获取x=c(2,3,5)的值并创建另一个数据帧。

3 个答案:

答案 0 :(得分:3)

我们可以使用%in%

a1 <- a[a$x %in% x,]

仅对列进行子集化&#39; x&#39;

a1 <- a[a$x %in% x, "x", drop=FALSE]

如果我们需要对列进行子集化&#39; x&#39;根据{{​​1}}向量

创建vector
x

答案 1 :(得分:2)

或者您可以使用subset

filter <- c(2,5)
subset(a, x %in% filter)

或等效地:

subset(a, match(x, filter, nomatch = 0)>0)

或者

a[match(a$x, filter, nomatch = 0)>0,]

   # x           y         z
# 2  2  0.76230930 0.9704342
# 5  5 -1.61846247 0.5786633
# 7  2  0.94024182 0.2805524
# 10 5 -0.08851427 0.6426568
# 12 2  0.78745436 0.1129637
# 15 5 -2.41274754 0.4826690
# 17 2 -0.37616238 0.9518877
# 20 5  1.18745381 0.8110062
# 22 2  0.03233245 0.4599623
# 25 5 -2.28360189 0.4836900

答案 2 :(得分:1)

我们可以使用%in%包中的值匹配函数filterdplyr动词(一个用于数据操作的优秀包)。

library(dplyr)
a1 <- data.frame(x = rep(1:5,5), y=rnorm(25), z=runif(25))
a2 <- filter(a1, x %in% c(2,3,5))

> a2
   x           y         z
1  2  0.28184946 0.3564756
2  3  0.05634123 0.9826746
3  5 -0.58611510 0.8119334
4  2  0.45211282 0.6267487
5  3 -0.64741961 0.7600619
6  5 -0.28781978 0.3216957
7  2  0.51440342 0.5165707
8  3  1.41958340 0.2328647
9  5 -0.27751501 0.5400576
10 2 -0.74835287 0.7976089
11 3  2.42364991 0.4141980
12 5  0.22175161 0.1051387
13 2  1.54876157 0.6408956
14 3  0.54940989 0.3968186
15 5 -1.16333440 0.9359615