Найти ближайшие предшествующие и последующие даты между фреймами данных

Question

У меня есть два следующих фрейма данных:

df1 <- data.frame(ID = c("A","A","B","B","C","D","D","D","E"),
             Date = as.POSIXct(c("2018-04-12 08:56:00","2018-04-13 11:03:00","2018-04-14 14:30:00","2018-04-15 03:10:00","2018-04-16 07:28:00","2018-04-17 11:17:00","2018-04-17 14:21:00","2018-04-18 09:56:00","2018-05-02 07:49:00")))

df2 <- data.frame(ID = c("A","A","A","B","C","D","D","D","D","D","E"),
              Date = as.POSIXct(c("2018-04-10 07:11:00","2018-04-11 18:59:00","2018-04-12 12:37:00","2018-04-15 01:43:00","2018-04-21 09:52:00","2018-04-15 20:25:00","2018-04-17 12:33:00","2018-04-17 14:21:00","2018-04-18 10:59:00","2018-04-20 14:11:00","2018-05-01 09:50:00")))

Для df1 я хотел бы сделать 2 вещи: во-первых, я хочу найти ближайшую предыдущую дату по идентификатору из df2.Во-вторых, я хочу найти ближайшую следующую дату по идентификатору из df2, снова без повторения значений.В обоих случаях я не хочу, чтобы даты из df2 повторялись в df1.

Использование функции roll = Inf из пакета data.table. Я могу объединить предыдущие даты по ID.

setDT(df1)
setDT(df2)

setkey(df1, ID, Date)
setkey(df2, ID, Date)[, PrecedingDate:=Date]

result <- df2[df1, roll=Inf]

Я не уверен, как можно перенести ближайшую следующую дату из df2 в df1 и как я могу убедиться, что даты не повторяются.

Результат должен быть следующим:

result <- data.frame(ID = c("A","A","B","B","C","D","D","D","E"),
                     Date = as.POSIXct(c("2018-04-12 08:56:00","2018-04-13 11:03:00","2018-04-14 14:30:00","2018-04-15 03:10:00","2018-04-16 07:28:00","2018-04-17 11:17:00","2018-04-17 14:21:00","2018-04-18 09:56:00","2018-05-02 07:49:00")),
                     PrecedingDate = as.POSIXct(c("2018-04-11 18:59:00","2018-04-12 02:37:00",NA,"2018-04-15 01:43:00",NA,"2018-04-15 20:25:00","2018-04-17 14:21:00",NA,"2018-05-01 09:50:00")),
                     FollowingDate = as.POSIXct(c("2018-04-12 02:37:00",NA,"2018-04-15 01:43:00",NA,"2018-04-21 09:52:00","2018-04-17 12:33:00","2018-04-17 14:21:00","2018-04-18 10:59:00",NA)))

Любая помощь здесь будет наиболее ценной.

Answer 1

Возможное решение с data.table :

df1[, PrecedingDate := df2[df1
                           , on = .(ID, Date <= Date)
                           , .(ID, Date = i.Date, pd = x.Date)
                           ][, .SD[.N], by = .(ID, Date)
                             ][shift(pd) == pd, pd := NA][, pd]
    ][, FollowingDate := df2[df1
                             , on = .(ID, Date >= Date)
                             , .(ID, Date = i.Date, fd = x.Date)
                             ][, .SD[1], by = .(ID, Date)][, fd]][]

, которое дает:

> df1
   ID                Date       PrecedingDate       FollowingDate
1:  A 2018-04-12 08:56:00 2018-04-11 18:59:00 2018-04-12 12:37:00
2:  A 2018-04-13 11:03:00 2018-04-12 12:37:00                <NA>
3:  B 2018-04-14 14:30:00                <NA> 2018-04-15 01:43:00
4:  B 2018-04-15 03:10:00 2018-04-15 01:43:00                <NA>
5:  C 2018-04-16 07:28:00                <NA> 2018-04-21 09:52:00
6:  D 2018-04-17 11:17:00 2018-04-15 20:25:00 2018-04-17 12:33:00
7:  D 2018-04-17 14:21:00 2018-04-17 14:21:00 2018-04-17 14:21:00
8:  D 2018-04-18 09:56:00                <NA> 2018-04-18 10:59:00
9:  E 2018-05-02 07:49:00 2018-05-01 09:50:00                <NA>

Это равно желаемомурезультат:

> all.equal(df1, as.data.table(result))
[1] TRUE

---

Используемые данные:

df1 <- data.frame(ID = c("A","A","B","B","C","D","D","D","E"),
                  Date = as.POSIXct(c("2018-04-12 08:56:00","2018-04-13 11:03:00","2018-04-14 14:30:00","2018-04-15 03:10:00","2018-04-16 07:28:00","2018-04-17 11:17:00","2018-04-17 14:21:00","2018-04-18 09:56:00","2018-05-02 07:49:00")))
df2 <- data.frame(ID = c("A","A","A","B","C","D","D","D","D","D","E"),
                  Date = as.POSIXct(c("2018-04-10 07:11:00","2018-04-11 18:59:00","2018-04-12 12:37:00","2018-04-15 01:43:00","2018-04-21 09:52:00","2018-04-15 20:25:00","2018-04-17 12:33:00","2018-04-17 14:21:00","2018-04-18 10:59:00","2018-04-20 14:11:00","2018-05-01 09:50:00")))
result <- data.frame(ID = c("A","A","B","B","C","D","D","D","E"),
                     Date = as.POSIXct(c("2018-04-12 08:56:00","2018-04-13 11:03:00","2018-04-14 14:30:00","2018-04-15 03:10:00","2018-04-16 07:28:00","2018-04-17 11:17:00","2018-04-17 14:21:00","2018-04-18 09:56:00","2018-05-02 07:49:00")),
                     PrecedingDate = as.POSIXct(c("2018-04-11 18:59:00","2018-04-12 12:37:00",NA,"2018-04-15 01:43:00",NA,"2018-04-15 20:25:00","2018-04-17 14:21:00",NA,"2018-05-01 09:50:00")),
                     FollowingDate = as.POSIXct(c("2018-04-12 12:37:00",NA,"2018-04-15 01:43:00",NA,"2018-04-21 09:52:00","2018-04-17 12:33:00","2018-04-17 14:21:00","2018-04-18 10:59:00",NA)))

Answer 2

Вот решение, использующее dplyr.Вы можете получить некоторые предупреждения для min max функций, но вы можете спокойно их игнорировать или подавлять.

library(dplyr)

closest_to_zero <- function(x) {
  neg <- which(x == max(x[x < 0]))
  pos <- which(x == min(x[x > 0]))
  c(previous = neg, following = pos)
}

result <- left_join(df1, df2, by = "ID") %>%
  group_by(ID, Date.x) %>%
  mutate(
    time_diff = Date.y - Date.x,
    Preceding = Date.y[closest_to_zero(time_diff)["previous"]],
    Following = Date.y[closest_to_zero(time_diff)["following"]]
  ) %>%
  distinct(ID, Date.x, Preceding, Following)

# A tibble: 9 x 4
# Groups:   ID, Date.x [9]
  ID    Date.x              Preceding           Following          
  <fct> <dttm>              <dttm>              <dttm>             
1 A     2018-04-12 08:56:00 2018-04-11 18:59:00 2018-04-12 12:37:00
2 A     2018-04-13 11:03:00 2018-04-12 12:37:00 NA                 
3 B     2018-04-14 14:30:00 NA                  2018-04-15 01:43:00
4 B     2018-04-15 03:10:00 2018-04-15 01:43:00 NA                 
5 C     2018-04-16 07:28:00 NA                  2018-04-21 09:52:00
6 D     2018-04-17 11:17:00 2018-04-15 20:25:00 2018-04-17 12:33:00
7 D     2018-04-17 14:21:00 2018-04-17 12:33:00 2018-04-18 10:59:00
8 D     2018-04-18 09:56:00 2018-04-17 14:21:00 2018-04-18 10:59:00
9 E     2018-05-02 07:49:00 2018-05-01 09:50:00 NA