hmh*_*sen 3 loops r vectorization
我想看看是否有办法矢量化我执行的计算。我搜索了这个答案,但找不到我需要的。
我有一个增长率向量。每个代表一个时期(在我的情况下为一年)。我想将此向量应用于某些本金。然后,在将第一个增长率应用于主体后,使用第一次迭代的结果并将第二个增长元素应用于新值。
这是一些用于复制的代码(全部在 中base
):
# Vector of interest or inflation rates
RateVector <- c(0.02, 0.03, 0.04, 0.05, 0.06, 0.05, 0.04, 0.03, 0.02, 0.01) # forecasted rates
Principal <- data.frame(Principal = 1000000) # actual value of indicator in most recent period as data frame (list)
Run Code Online (Sandbox Code Playgroud)
这是我尝试矢量化:
sapply(Principal, "*", 1 + cumsum(RateVector))
Run Code Online (Sandbox Code Playgroud)
问题在于该sapply
函数不会保存新金额,而是将利率向量应用于相同的初始本金。这实际上是我对这段代码的期望。我不知道如何在每次迭代后从元素到元素保存新值。
这就是我使用循环解决问题的方法:
AmountVector <- Principal # initialize output vector
# Compound growth growth calculation loop
for(i in 1:length(RateVector)){
Principal = Principal * (1 + RateVector)[i]
AmountVector <- rbind(AmountVector,Principal)
}
# Print result
AmountVector
Run Code Online (Sandbox Code Playgroud)
这是一个“累积产品”,因此?cumprod
您需要:
1000000 * cumprod(1 + RateVector)
# [1] 1020000 1050600 1092624 1147255 1216091 1276895 1327971 1367810 1395166
#[10] 1409118
cbind(AmountVector, newresult = 1000000 * c(1, cumprod(1 + RateVector)))
# Principal newresult
#1 1000000 1000000
#2 1020000 1020000
#3 1050600 1050600
#4 1092624 1092624
#5 1147255 1147255
#6 1216091 1216091
#7 1276895 1276895
#8 1327971 1327971
#9 1367810 1367810
#10 1395166 1395166
#11 1409118 1409118
Run Code Online (Sandbox Code Playgroud)