ведение оперативной записи предыдущего уникального значения - dplyr в data.table

Question

Я пытаюсь преобразовать функцию dplyr, которую я имею, в data.table, и мне сложно добиться четности. Любая помощь высоко ценится!

Данные

DT = data.table(
  "brk" = 1L,
  "est" = c(900, 900, 1000, 1000, 1100),
  "t" = c(0, 12, 15, 22, 30)
)

желаемый результат

Desired_DT = data.table(
  "brk" = 1L,
  "est" = c(900, 900, 1000, 1000, 1100),
  "t" = c(0, 12, 15, 22, 30),
  "prev_est" = c(NA, NA , 900, 900, 1000),
  "prev_est_age" = c(NA, NA, 15, 22, 15)
)

один из возможных способов dplyr

DT %>% 
  left_join(
    DT %>% 
      group_by(est) %>% 
      slice(1) %>% 
      ungroup() %>% 
      mutate(
        prev_est = lag(n = 1L, est),
        prev_t = lag(n = 1L, t)
      )
  ) %>% tidyr::fill(.data$prev_est, .data$prev_t) %>% 
  mutate(prev_est_age = t - prev_t)

Answer 1

Вот опция, которая выполняет синтаксис, аналогичный приведенному в таблице.

library(data.table)
library(zoo)
# get the first row, grouped by 'est', create two lag columns of 'est', t'
dt1 <- DT[, head(.SD, 1), est][, c("prev_est", "prev_t") := 
        shift(.SD), .SDcols = c('est', 't')][] 
# do a join on 'brk', 'est', 't', do the `fill` the `na.locf0` from `zoo`
# and lastly create the column 'prev_est_age' as the difference of t, prev_t
dt1[DT, on = .(brk, est, t)][, c("prev_est", "prev_t") := lapply(.SD, 
  zoo::na.locf0), .SDcols = c("prev_est", "prev_t")][,
      prev_est_age := t - prev_t][]

---

Мы также можем использовать подвижное соединение

dt1 <- unique(DT, by = 'est')[,  paste0("prev_", names(DT)[2:3]) := 
           shift(.SD), .SDcols = est:t][]
out <- dt1[DT, on = .(brk, est, t), roll = Inf][, 
                prev_est_age := t - prev_t][, prev_t := NULL][]
identical(Desired_DT, out)
#[1] TRUE