R data.table 使用日期非等值连接与查找 data.table 从查找中添加列

Question

我想使用日期查找向 data.table 添加一列，匹配大于或等于 data.table 中日期的查找行。

例如，给定这个包含四个日期的简短表格:

dt1<-data.table(Dt=as.Date(rbind("2023-01-01", "2023-01-02", "2023-01-14", "2023-01-15")))
dt1
           Dt
1: 2023-01-01
2: 2023-01-02
3: 2023-01-14
4: 2023-01-15

并给出第二个日期查找表:

dtLookup<-data.table(Dt=as.Date(rbind("2023-01-07", "2023-01-14", "2023-01-21")))
dtLookup
           Dt
1: 2023-01-07
2: 2023-01-14
3: 2023-01-21

我想以某种方式加入并最终得到原始 dt1 表，该表具有与查找日期相对应的附加列。所以在这个例子中 dt1 看起来像这样:

dt1
           Dt   LookupDt
1: 2023-01-01 2023-01-07
2: 2023-01-02 2023-01-07
3: 2023-01-14 2023-01-14
4: 2023-01-15 2023-01-21

有没有办法将 dt1 非等值连接到 dtLookup 并根据 dtLookup 中找到的行设置新的 dt1 列？

Answer 1

我认为 findInterval (如 akrun 的答案)可能是给定示例数据的最佳选择，但如果您需要连接(例如其他列)，请尝试这个:

dtLookup[, Dt2 := shift(Dt, type = "lag", fill = min(dt1$Dt)-1)]
dtLookup
#            Dt        Dt2
#        <Date>     <Date>
# 1: 2023-01-07 2022-12-31
# 2: 2023-01-14 2023-01-07
# 3: 2023-01-21 2023-01-14
dt1[dtLookup, Lookup := i.Dt, on = .(Dt > Dt2, Dt <= Dt)]
dt1
#            Dt     Lookup
#        <Date>     <Date>
# 1: 2023-01-01 2023-01-07
# 2: 2023-01-02 2023-01-07
# 3: 2023-01-14 2023-01-14
# 4: 2023-01-15 2023-01-21

这会在 dtLookup 中生成一个我们可以在内部使用的范围。