在R中递归添加一列

发布于05月05日

我有一个dataframe，df，看起来是这样的:

Year Colombia    Chile
------------------------
2006   9.5660    24.5576
2007  26.0079    10.9652
2008  16.0893    2.3721
2009  -5.1036   -5.0465

我想加上两列index_Colombia和index_Chile，第一行f(1)的值是index_Colombia，第二行Colombia的值是index_Colombia*(1+26.0079/index_Colombia)，智利的值是index_Colombia*(1+10.9652/index_Colombia)，依此类推...这样我们就得到了下面的df

Year Colombia    Chile    index_Colombia   index_Chile 
-------------------------------------------------------
2006   9.5660    24.5576      100            100       
2007  26.0079    10.9652   126.0079         110.9652
2008  16.0893    2.3721     146.2816         113.5974
2009  -5.1036   -5.0465     138.8159        107.864

我突然想到，这是有必要的mutate函数，但我不知道如何制定它....你能帮助我吗？

推荐答案

一个可能的解决方案:

library(dplyr)

df <- data.frame(
  Year = c(2006L, 2007L, 2008L, 2009L),
  Colombia = c(9.566, 26.0079, 16.0893, -5.1036),
  Chile = c(24.5576, 10.9652, 2.3721, -5.0465)
)

bind_rows(df[1,],
 df %>% slice(-1) %>% 
 mutate(across(-Year, ~ 100 * cumprod(1 + .x/100), .names = "index_{.col}"))) %>% 
 replace(is.na(.), 100)

#>   Year Colombia   Chile index_Colombia index_Chile
#> 1 2006   9.5660 24.5576       100.0000    100.0000
#> 2 2007  26.0079 10.9652       126.0079    110.9652
#> 3 2008  16.0893  2.3721       146.2817    113.5974
#> 4 2009  -5.1036 -5.0465       138.8161    107.8647