使用R代碼的條件變量

Question

我有一個名爲「dats」的數據集。

id y i j 
    1 0 1 1 
    1 0 1 2 
    1 0 1 3 
    2 1 2 1 
    2 1 2 2 
    2 1 2 3 

我想計算，一個新的變量ynew基於=（yij-1 * yij）（Y11 Y12 *，Y12 Y13 * ....等等）。我曾嘗試了這種方式：

ynew <- NULL 

for(p in 1) 
    { 
    for (q in ni) 
     { 
     ynew[p,q] <- dats$y[dats$i==p & dats$j==q-1]*dats$y[dats$i==p & dats$j==q] 
     } 
    } 

ynew

但它顯示的錯誤！

預計輸出

id y i j ynew 
    1 0 1 1 NA 
    1 0 1 2 0 
    1 0 1 3 0 
    2 1 2 1 NA 
    2 1 2 2 1 
    2 1 2 3 1 

可能有人幫助嗎？ TIA

Answer 1

可能是我們需要的只是用的lag 'Y' 的 '身份證'

library(data.table) 
setDT(dats)[, ynew := y * shift(y), by = id] 
dats 
# id y i j ynew 
#1: 1 0 1 1 NA 
#2: 1 0 1 2 0 
#3: 1 0 1 3 0 
#4: 2 1 2 1 NA 
#5: 2 1 2 2 1 
#6: 2 1 2 3 1 

它也與roll_prod

library(RcppRoll) 
setDT(dats)[, ynew := c(NA, roll_prod(y, 2)), by = id] 
dats 
# id y i j ynew 
#1: 1 0 1 1 NA 
#2: 1 0 1 2 0 
#3: 1 0 1 3 0 
#4: 2 1 2 1 NA 
#5: 2 1 2 2 1 
#6: 2 1 2 3 1 

Answer 2

使用dplyr和rollapply從進行分組乘zoo包，

library(dplyr) 
library(zoo) 

dats %>% 
    group_by(id) %>% 
    mutate(ynew = c(NA, rollapply(y, 1, by = 2, prod))) 

#Source: local data frame [6 x 5] 
#Groups: id [2] 

#  id  y  i  j ynew 
# (int) (int) (int) (int) (dbl) 
#1  1  0  1  1 NA 
#2  1  0  1  2  0 
#3  1  0  1  3  0 
#4  2  1  2  1 NA 
#5  2  1  2  2  1 
#6  2  1  2  3  1