data.table

Question

舉一個簡單的例子，我需要創建下表中的R列的遞推更新（這是我想要的結果）：

library(data.table) 
DT <- data.table(A= c(2,5,4,-2,4), 
        B= c(1000,1100,1375,1650,1485), 
        C= c(50,55,68.75,82.5,74.25), 
        D= c(100,275,275,-165,297)) 
DT 

這是我的嘗試，到目前爲止這是不工作：

DT.2 <- data.table(A= c(2,5,4,-2,4)) 
DT.2[, B := 1000 ] # B should begin at 1000, then cumulatively add column D 
DT.2[, C := B * 0.05 ] 
DT.2[, D := A * C ] 
DT.2[, B := 1000 + cumsum(shift(D, type= "lag", fill=FALSE)) ] 
DT.2 

正如您所看到的，列不會正確更新，因爲每列都依賴於其他列的結果。在Excel中這是一個非常簡單的計算，我只需要了解如何使其適應R. 謝謝

Answer 1

就我所知，您需要一個循環。這裏是我的輸入數據：

DT <- data.table(A = c(2, 5, 4, -2, 4), 
       B = c(1000, rep(NA, 4)), 
       C = numeric(5), 
       D = numeric(5)) 

，循環使用：

#initial row 
DT[1, c("C", "D") := .(.05 * B, .05 * A * B)] 

#subsequent rows 
for (nn in 2:nrow(DT)){ 
    new_B <- DT[nn - 1L, B + D] 
    DT[nn, c("B", "C", "D") := .(new_B, .05 * new_B, .05 * A * new_B)] 
} 

也很容易被轉換爲sapply，所以你不必nn坐在你的命名空間周圍（但現在它會打印無意義的東西到控制檯）：

sapply(2:nrow(DT), function(nn){ 
    (same as before)}) 

是的，我能想象這將在Excel中看起來更簡單，因爲它自動更新爲爲你點擊+拖動鼠標。

Answer 2

我認爲你真正想要的功能，我猜測，它將使更多類似這樣的感覺：

fun <- function(id, ## not sure what this is, probably an id 
       start = 1000, 
       rate = .05, 
       gain_or_loss){ 
    require(data.table) 

    cnt <- length(id) 
    ret <- data.table(ID = c(id, NA), 
         bal = cumsum(c(start, gain_or_loss)), 
         fixed_change = rep(NA_real_, cnt + 1), 
         total_change = c(gain_or_loss, NA_real_)) 
    ret[ , fixed_change := bal * 0.05 ] 
    ret <- ret[] ## needed because of a data.table quirk 
    return(ret) 
} 

這將使：

fun(id = c(2,5,4,-2,4), 
    gain_or_loss = c(100,275,275,-165,297)) 

# ID bal fixed_change total_change 
# 1: 2 1000  50.00   100 
# 2: 5 1100  55.00   275 
# 3: 4 1375  68.75   275 
# 4: -2 1650  82.50   -165 
# 5: 4 1485  74.25   297 
# 6: NA 1782  89.10   NA 

我還想以後添加代碼require說if (length(id) != length(gain_or_loss)) stop("id and gain_or_loss need to be of the same length")