r - for 循环中的超前和滞后

Question

我的数据框 crisisdata 由许多向量组成:年、国家、a、b、c

我想在 for 循环中为 a、b 和 c 创建滞后和引导:

我的尝试:

crisis_variables <- c("a", "b", "c")

for (var in crisis_variables){
  
  # add lags
  crisisdata[, (paste0("l",1:4, "_", var)):= shift(as.name(var),1:4), by=country]

  # add leads
  crisisdata[, (paste0("f",0:4, "_", var)):= shift(as.name(var),0:-4), by=country]

但是在每个滞后变量和超前变量中我只有 NA 值。

Answer 1

您可以使用高度灵活的 shift 函数来一次生成所有滞后和超前，如下所示:

# create column names
n = -4:4
cols = paste0(ifelse(n>0, "l", "f"), abs(n))
cols = paste0(cols, "_", rep(crisis_variables, each=length(cols)))   
# f4_a, f3_a, ..., l3_c, l4_c

# generate all lags and leads
crisisdata[, (cols) := shift(.SD, n), by=country, .SDcols=crisis_variables]