我有一个数据集(mydata),其中包含多个列,这些列可以适合存储在另一个数据集(mycomparison)中的范围内.
我想加入mycomparison到mydata了哪里mydata值都在范围内mycomparison.
library(data.table)
mydata<-data.table(
id=1:5,
val1=seq(10000, 50000, by=10000),
val2=floor(rnorm(5,mean=400,sd=100)),
val3=rnorm(5,mean=.7,sd=.1)
)
mycomparison<-data.table(
Name=LETTERS[1:3],
minval1=c(0,30000,10000),
maxval1=c(50000,80000,30000),
minval2=c(300,400,300),
maxval2=c(800,800,800),
minval3=c(0,.5,.2),
maxval3=c(1,.9,.8),
correspondingval=c(.1,.2,.3)
)
Run Code Online (Sandbox Code Playgroud)
> mydata.withmatches
id val1 val2 val3 Name minval1 maxval1 minval2 maxval2 minval3 maxval3 correspondingval
1: 1 10000 387 0.4844319 A 0 50000 300 800 0 1 0.1
2: 2 20000 425 0.7856313 NA NA NA NA NA NA NA NA
3: 3 30000 324 0.8063969 NA NA NA NA NA NA NA NA
4: 4 40000 263 0.5590113 NA NA NA NA NA NA NA NA
5: 5 50000 187 0.8764396 NA NA NA NA NA NA NA NA
Run Code Online (Sandbox Code Playgroud)
这感觉/非常笨重,涉及交叉连接数据(使用optiRum::CJ.dt),进行大的逻辑检查,然后重新组装数据.
library(optiRum)
workingdt<-CJ.dt(mydata,mycomparison)
matched<-workingdt[val1>=minval1 &
val1<=maxval1 &
val2>=minval2 &
val2<=maxval2 &
val3>=minval3 &
val3<=maxval3][which.min(correspondingval)]
notmatched<-mydata[id!= matched[,id]]
all<-list(matched,notmatched)
mydata.withmatches<- rbindlist(all, fill=TRUE, use.names=TRUE)
Run Code Online (Sandbox Code Playgroud)
我知道foverlaps但它会在一个区间内工作,而不是像在这种情况下的许多范围.
我希望有一个不那么笨重,更优雅的解决方案.
好问题!您可以在下面找到我的快速解决方案,但对于我的口味来说,它仍然有点太笨重。
生成的mydata集合:
id val1 val2 val3
1: 1 10000 377 0.7912443
2: 2 20000 378 0.7709792
3: 3 30000 484 0.7049517
4: 4 40000 513 0.5169590
5: 5 50000 474 0.7987448
Run Code Online (Sandbox Code Playgroud)
一个简单的函数,用于过滤掉比较数据集中不匹配的行(可以匹配多行)。
library(dplyr)
find_interval_func<-function(var.min.name, var.max.name, value, val.to.return){
compset<-data.frame(mycomparison)
for(i in 1:length(var.min.name)){
compset<-
compset %>%
filter_(paste0(var.min.name[[i]], "<=", value[[i]]),
paste0(var.max.name[[i]], ">", value[[i]]))
}
paste(compset[,val.to.return], collapse="|")
}
Run Code Online (Sandbox Code Playgroud)
结果:
> mydata %>%
+ group_by(1:n()) %>%
+ mutate(matchedValue = find_interval_func(c("minval1", "minval2", "minval3"),
+ c("maxval1", "maxval2", "maxval3"),
+ c(val1, val2, val3),
+ "Name"))
Source: local data table [5 x 6]
id val1 val2 val3 1:n() matchedValue
1 1 10000 377 0.7912443 1 A|C
2 2 20000 378 0.7709792 2 A|C
3 3 30000 484 0.7049517 3 A|B
4 4 40000 513 0.5169590 4 A|B
5 5 50000 474 0.7987448 5 B
Run Code Online (Sandbox Code Playgroud)