data.table R 中的子集 ID 和日期

Question

我有一个类似于我创建的下一个示例的大矩阵（我有 70 列和数百万行）：

a <- seq(as.IDate("2011-12-30"), as.IDate("2014-01-04"), by="days")
data <- data.table(ID = 1:length(a), date1 = a)

我想提取 ID 中的所有行，它包含个人的 ID，以及我需要从那个人那里提取的日期。 一个人可以有多条线路。

a <- seq(as.IDate("2011-12-30"), as.IDate("2014-01-04"), by="week")
b <- seq(as.IDate("2012-01-01"), as.IDate("2014-01-06"), by="week")
IDs <- data.table(ID = 1:length(a), date1 = a, date2 = b)

目前，我的解决方案不是很快，什么会更好？

A <- list()
for(i in 1:dim(IDs)[1]){  
 A[[i]] <- data[ID == IDs[i,ID] & (date1 %between% IDs[i,.(date1,date2)]),]
}

Answer 1

我认为您正在寻找非 equi 内部联接：

IDs[data, on=.(ID, date1<=date1, date2>=date1), nomatch=0L, .(ID, date1=i.date1)]

或关联地，

data[IDs, on=.(ID, date1>=date1, date1<=date2), nomatch=0L, .(ID, date1=x.date1)]

或将其视为非 equi 半联接：

data[IDs[data, on=.(ID, date1<=date1, date2>=date1), nomatch=0L, which=TRUE]]

output：

   ID      date1
1:  1 2011-12-30