r - 从 R 中的先前观察中查找信息

Question

这个问题在这里已经有了答案:





R Add previous month value as a column

(2 个回答)


6年前关闭。




我有一个包含交易的数据表。除其他事项外，交易标有日期/时间和客户。

library(data.table)
transactions <- data.table(trans.id = 1:10,
                           customer.id = c(rep(c('a','b'),3),'c'))
set.seed(42)
transactions[, time := sample(500, 10, T)]

对于每笔交易，我想查找客户的前一笔交易，这样我就可以计算自上次交易以来的时间。这是我能做到的最好的...

Find.Prior.ID <- function(id.curr) {
  trans.curr <- transactions[trans.id == id.curr]
  time.curr <- trans.curr[, time]
  customer <- trans.curr[, customer.id]
  prior.trans <- transactions[
    time < time.curr & customer.id == customer]
  if (dim(prior.trans)[1] == 0) {return(NA)}
  time.prior <- prior.trans[,max(time)]
  id.prior <- prior.trans[
    time == time.prior,
    trans.id]
  return(id.prior)
}

transactions[, prior.id := unlist(lapply(
  X = trans.id, FUN = Find.Prior.ID))]

一旦我有 prior.id填充一个简单的合并允许我映射我想要的任何其他信息。我的问题真的归结为运行时间。这种结构的运行时间很长。有没有办法更快地做到这一点？

答案 (特定于相关示例)

setorder(transactions, time)
transactions[, prior.id:=c(NA,trans.id[-.N]), by=customer.id]

Answer 1

在 data.table版本1.9.5有一个新的shift接受参数的函数 type = lag用于计算滞后变量。

transactions[, prior.id := shift(trans.id, type = "lag"), by = customer.id]