根据 R IV 中每个 ID 的 eventdate 填写订单

Question

我想知道如何创建一个名为“order”的新列，该列会自动填充 integer (-3:3) 的序列。 该序列以“eventdate”为中心，因此，在“eventdate”，order 的值为零。

我有下表。

df<-structure(list(Date = structure(c(15047L, 15048L, 15049L, 15050L, 
                                  15051L, 15052L, 15053L, 15054L, 15142L, 15143L, 15144L, 15145L, 
                                  15146L, 15147L, 15148L, 15119L, 15120L, 15121L, 15122L, 15123L, 
                                  15124L, 15125L, 15126L, 15497L, 15498L, 15499L, 15500L, 15501L, 
                                  15502L, 15503L, 15504L, 15505L, 15506L), class = c("IDate", "Date"
                                  )), ID = c(1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 
                                             1L, 1L, 2L, 2L, 2L, 2L, 2L, 2L, 2L, 2L, 3L, 3L, 3L, 3L, 3L, 3L, 
                                             3L, 3L, 3L, 3L), eventdate = structure(c(NA, NA, NA, NA, 15051L, 
                                                                                      NA, NA, NA, NA, NA, NA, 15145L, NA, NA, NA, NA, NA, NA, 15122L, 
                                                                                      NA, NA, NA, NA, NA, NA, NA, NA, 15501L, NA, 15503L, NA, NA, NA
                                             ), class = c("IDate", "Date")), Value = c(10L, 11L, 12L, 11L, 
                                                                                       15L, 17L, 18L, 15L, 20L, 21L, 22L, 25L, 26L, 25L, 26L, 5L, 6L, 
                                                                                       7L, 9L, 10L, 11L, 15L, 16L, 20L, 22L, 23L, 24L, 28L, 29L, 25L, 
                                                                                       24L, 26L, 24L)), row.names = c(NA, -33L), class = c("data.table", 
                                                                                                                                           "data.frame"))

同样，我想创建一个名为“order”的新列，以便在 eventdate 不是 NA 的那一天，它的 order 为零。 然后，我想将负连续递减整数填充到前一行，将正连续递增整数填充到每个 ID 的下面行。 同样，我只需要事件日期的前三行和事件日期之后的三行，这使得每个事件日期有七行，因为这七行包括事件日期本身。 请注意，每个 ID 可能有多个事件日期。

另外，请注意 ID 3 的两个事件日期发生得太频繁。 两个活动日期之间只有一天。 在这种情况下，我想忽略即将到来的 eventdate (2012-06-12)，并创建三个连续的递增顺序整数，就好像 2012-06-12 没有发生任何事情一样。 同样，在 2012 年 6 月 12 日，我想忽略 eventdate(2012-06-10) 并在 2012 年 6 月 12 日之前创建三个连续的降序整数。 比较当前表和所需的 output 会更有意义。

下面是所需的 output。

output<-structure(list(Date = structure(c(15048L, 15049L, 15050L, 15051L, 
                                  15052L, 15053L, 15054L, 15142L, 15143L, 15144L, 15145L, 15146L, 
                                  15147L, 15148L, 15119L, 15120L, 15121L, 15122L, 15123L, 15124L, 
                                  15125L, 15498L, 15499L, 15500L, 15501L, 15502L, 15503L, 15504L, 
                                  15500L, 15501L, 15502L, 15503L, 15504L, 15505L, 15506L), class = c("IDate", 
                                                                                                     "Date")), ID = c(1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 1L, 
                                                                                                                      1L, 1L, 1L, 2L, 2L, 2L, 2L, 2L, 2L, 2L, 3L, 3L, 3L, 3L, 3L, 3L, 
                                                                                                                      3L, 3L, 3L, 3L, 3L, 3L, 3L, 3L), eventdate = structure(c(NA, 
                                                                                                                                                                               NA, NA, 15051L, NA, NA, NA, NA, NA, NA, 15145L, NA, NA, NA, NA, 
                                                                                                                                                                               NA, NA, 15122L, NA, NA, NA, NA, NA, NA, 15501L, NA, NA, NA, NA, 
                                                                                                                                                                               NA, NA, 15503L, NA, NA, NA), class = c("IDate", "Date")), Value = c(11L, 
                                                                                                                                                                                                                                                   12L, 11L, 15L, 17L, 18L, 15L, 20L, 21L, 22L, 25L, 26L, 25L, 26L, 
                                                                                                                                                                                                                                                   5L, 6L, 7L, 9L, 10L, 11L, 15L, 22L, 23L, 24L, 28L, 29L, 25L, 
                                                                                                                                                                                                                                                   24L, 24L, 28L, 29L, 25L, 24L, 26L, 24L), order = c(-3L, -2L, 
                                                                                                                                                                                                                                                                                                      -1L, 0L, 1L, 2L, 3L, -3L, -2L, -1L, 0L, 1L, 2L, 3L, -3L, -2L, 
                                                                                                                                                                                                                                                                                                      -1L, 0L, 1L, 2L, 3L, -3L, -2L, -1L, 0L, 1L, 2L, 3L, -3L, -2L, 
                                                                                                                                                                                                                                                                                                      -1L, 0L, 1L, 2L, 3L)), row.names = c(NA, -35L), class = c("data.table", 
                                                                                                                                                                                                                                                                                                                                                                "data.frame"))

非常感谢您！

Answer 1

我们可以在创建行序列列后filter 'eventdate'中具有非NA值的元素，然后在基于序列（ -3:3 ）扩展后使用序列列从原始'df'中子集对应的列值

library(dplyr)
library(tidyr)
out2 <- df %>%
    mutate(rn = row_number()) %>% 
    filter(complete.cases(eventdate)) %>% 
    rowwise %>%
    mutate(order = list(-3:3), rn = list(rn + order)) %>%
    ungroup %>%
    unnest(where(is.list)) %>% 
    mutate(across(c("Date", "eventdate", "Value"),
         ~ df[[cur_column()]][rn])) %>% 
    select(-rn) %>% 
    mutate(eventdate = case_when(order == 0 ~ eventdate))

-检查预期的 output

> all.equal(out2, output, check.attributes = FALSE)
[1] TRUE