使用R选择从哪里开始的程序

时间:2013-08-14 12:44:46

标签: r data.table plyr sqldf

我有两个数据帧,这是第一个:

df <- data.frame(p=letters[1:3],y1=c(2,4,3))
df
  p y1
1 a 2
2 b 4
3 c 3

和第二个:

df2 <- data.frame(p=rep(letters[1:3],c(3,2,4)),y2=c(3,1,1,4,3,4,3,3,1),d=rep(1,length=9))
df2
  p y2 d
1 a 3 1
2 a 1 1
3 a 1 1
4 b 4 1
5 b 3 1
6 c 4 1
7 c 3 1
8 c 3 1
9 c 1 1

我想要做的是在df2中获取这些行,p(a,b,c等)的每个值,其中d=1(在这种情况下是所有行),y2大于y1p分组的df

因为这个解释可能没有意义,需要踢的两行:df2中的第1行,因为a,{{1} }大于y2=3中的y1=2第6行,因为对于df中的cdf2的值为4,但yc的值为3。

由于我正在使用data.tables,“data.table-solution”会很好,可能是这样的:

df

3 个答案:

答案 0 :(得分:6)

您应该在子集之前使用merge

使用data.table

library(data.table)
merge(data.table(df1,key='p'),
      data.table(df2,key='p'))[d==1 & y2 > y1]
   p y1 y2 d
1: a  2  3 1
2: c  3  4 1

使用base merge

subset(merge(df1,df2), d==1 & y2 > y1)
  p y1 y2 d
1 a  2  3 1
6 c  3  4 1

修改

对于data.table解决方案,最好使用连接Y [X],使用X的键查找Y的行。(LEFT OUTER JOIN)

DF2 <- data.table(df2,key='p')
DF1 <- data.table(df1,key='p')
DF2[DF1][d==1 & y2 > y1]

   p y2 d y1
1: a  3 1  2
2: c  4 1  3

答案 1 :(得分:1)

与rmk类似但使用plyr:

library(plyr)
dfa <- data.frame(p=letters[1:3],y1=c(2,4,3))
dfa

dfb <- data.frame(p=rep(letters[1:3],c(3,2,4)),y2=c(3,1,1,4,3,4,3,3,1),d=rep(1,length=9))
dfb

dfb <- join(dfa, dfb, by = "p", type = "left", match = "all")
dfb

dfb$z <- ifelse(dfb$y2>dfb$y1, 1, 0)
dfb[dfb$z==1, ]

答案 2 :(得分:0)

尝试:

df3 <- merge(df,df2,by=1)
> df3
  p y1 y2 d
1 a  2  3 1
2 a  2  1 1
3 a  2  1 1
4 b  4  4 1
5 b  4  3 1
6 c  3  4 1
7 c  3  3 1
8 c  3  3 1
9 c  3  1 1


> df3[df3$y2>df3$y1 & df3$d==1,]
  p y1 y2 d
1 a  2  3 1
6 c  3  4 1