通过加入R的条件组

时间:2015-06-04 14:28:27

标签: r join conditional

我是R的新手,而且因以下问题而烦恼。我有两个日期向量(向量不一定是对齐的,也不是相同的长度)。

我想为第一个向量中的每个日期找到第二个向量中的下一个日期。

vecA <- as.Date(c('1951-07-01', '1953-01-01', '1957-04-01', '1958-12-01',
       '1963-06-01', '1965-05-01'))
vecB <- as.Date(c('1952-01-12', '1952-02-01', '1954-03-01', '1958-08-01', 
           '1959-03-01', '1964-03-01', '1966-05-01'))

在SQL中我会写这样的东西,但我在SO中找不到关于如何在R中执行此操作的任何提示。

 select vecA.Date, min(vecB.Date)
       from vecA inner join vecB
       on vecA.Date < vecB.Date
       group by vecA.Date

输出应如下所示:

Start      End
1951-07-01 1952-01-12
1953-01-01 1954-03-01
1957-04-01 1958-08-01
1958-12-01 1959-03-01
1963-06-01 1964-03-01
1965-05-01 1966-05-01

2 个答案:

答案 0 :(得分:2)

这是使用data.table滚动连接

的可能解决方案
library(data.table)
dt1 <- as.data.table(vecA) ## convert to `data.table` object
dt2 <- as.data.table(vecB) ## convert to `data.table` object
setkey(dt2) # key in order to perform a binary join
res <- dt2[dt1, vecB, roll = -Inf, by = .EACHI] # run the inner join while selecting closest date 
setnames(res, c("Start", "End"))
res
#         Start        End
# 1: 1951-07-01 1952-01-12
# 2: 1953-01-01 1954-03-01
# 3: 1957-04-01 1958-08-01
# 4: 1958-12-01 1959-03-01
# 5: 1963-06-01 1964-03-01
# 6: 1965-05-01 1966-05-01

或者,我们也可以这样做:

data.table(vecA=vecB, vecB, key="vecA")[dt1, roll=-Inf]

答案 1 :(得分:1)

此代码将按照您的要求执行,但不清楚您要完成的操作,因此这可能不是最好的方法。本质上,此代码首先对两个向量进行排序,以确保它们处于相同的排序中。然后,使用for循环,它遍历vecA中的所有元素,并使用x < vecB找出vecB中哪些元素小于x

它包含在which中,它返回向量的每个TRUE元素的数字索引,然后在min中,它给出最小的数字索引。然后将其用于子集vecB以返回日期;它全部包含在print中,因此您可以看到循环的输出。

这可能不是最好的方法,但如果没有更多关于目标的背景信息,至少应该让你开始。

> vecA <- vecA[order(vecA)]
> vecB <- vecB[order(vecB)]
> for(x in vecA) {print(vecB[min(which(x < vecB))])}
[1] "1952-01-12"
[1] "1954-03-01"
[1] "1958-08-01"
[1] "1959-03-01"
[1] "1964-03-01"
[1] "1966-05-01"