我使用了稍微改变的样本数据,以确保彼此相隔
样本数据
names <- c("John", "John", "Rick", "Rick", "Katie", "Katie", "Harry", "Harry" )
date1 <- c("1-3-2016", "28-4-2020", "13-1-2018", "4-2-2020", "5-1-2019", "29-1-2020", "27-8-2018", "4-2-2020")
date2 <- c("16-4-2020", "28-5-2020", "2-3-2020", "16-2-2020", "25-2-2020", "10-4-2020", "27-6-2019", "8-4-2020")
df1 <- data.frame(names,date1,date2)
names date1 date2
1 John 1-3-2016 16-4-2020
2 John 28-4-2020 28-5-2020 # !! <-- altered so interval-gap with line 1 <= 60 days
3 Rick 13-1-2018 2-3-2020
4 Rick 4-2-2020 16-2-2020
5 Katie 5-1-2019 25-2-2020
6 Katie 29-1-2020 10-4-2020
7 Harry 27-8-2018 27-6-2019
8 Harry 4-2-2020 8-4-2020
names <- c("John", "Rick", "Katie", "Harry", "Harry")
date1 <- c("1-3-2016", "13-1-2018", "5-1-2019", "27-8-2018", "4-2-2020")
date2 <- c("28-5-2020", "2-3-2020", "10-4-2020", "27-6-2019", "8-4-2020")
df2 <- data.frame(names,date1,date2)
names date1 date2
1 John 1-3-2016 28-5-2020 # !! <-- joined, since gap <= 60 days
2 Rick 13-1-2018 2-3-2020 # !! <-- fixed type in your sample data provided
3 Katie 5-1-2019 10-4-2020
4 Harry 27-8-2018 27-6-2019
5 Harry 4-2-2020 8-4-2020
代码
代码使用了data.table 和intervals 包。由于intervals 仅适用于数字(整数或实数)间隔,日期列在间隔创建/扩展/合并之前转换为数字,并在处理后返回日期格式。
下面的代码做了什么:
- Loop (
lapply() over cucks of split (using data.table::split() with the by-argument. ,按名称。使用keep.by = FALSE,因为我们不需要它,并且别名也存储在创建列表的名称。
- 对于每个 cunck (=name),根据两个日期列定义间隔,并将这些间隔扩展为
gap 变量值的一半(在下面的代码中设置为 60)。因此,每个间隔都会在前后延长 30 天。
然后它加入重叠/接触(扩展)间隔,最后删除扩展。
- 使用
data.table::rbindlist(),将所有结果重新合并在一起。
- 设置列名,并将数值日期转换回数据格式
.
library( data.table )
library( intervals )
#set maximum gap between intervals
gap = 60
#set data to data.table format
setDT(df1)
#set dates to numeric (required by the intervals-package)
df1[, c("date1", "date2") := lapply( .SD, as.numeric ), .SDcols = c("date1", "date2") ]
#where the magic happens (see text above for explanation )
ans <- data.table::rbindlist(
lapply( split( df1 , by = "names", keep.by = FALSE ), function(x) {
as.data.table(
intervals::close_intervals( intervals::contract( intervals::reduce( intervals::expand(
intervals::Intervals( x, type = "Z" ),
gap/2 ) ), gap/2 )
)
)
}),
use.names = TRUE, idcol = "name" )
#use names from df1
setnames( ans, names(ans), names(df1) )
#set numeric back to date
ans[, c("date1", "date2") := lapply( .SD, as.Date, origin = "1970-01-01" ), .SDcols = c("date1", "date2") ]
输出
names date1 date2
1: John 2016-03-01 2020-05-28
2: Rick 2018-01-13 2020-03-02
3: Katie 2019-01-05 2020-04-10
4: Harry 2018-08-27 2019-06-27
5: Harry 2020-02-04 2020-04-08