R语言使用dplyr::mutate基于同列前序值计算后续值的实现方法

实现方法 #

你可以通过cumprod（累积乘积函数）和dplyr::mutate结合实现，无需引入额外循环，代码如下：

# 加载dplyr包
library(dplyr)

df <- df %>%
  mutate(
    size = first(size) * cumprod(c(1, tail(1 + growth, -1)))
  )

逻辑说明

• first(size)取2018年的初始size值100
• tail(1 + growth, -1)提取从2019年开始的各年(1+增长率)序列
• 序列开头补1，保证2018年的计算结果就是初始值，再通过cumprod计算累积乘积，和初始值相乘就得到各年的size

如果需要更贴合递推逻辑的写法，也可以搭配purrr::accumulate函数实现，这个函数天生支持依赖上一步结果的迭代计算：

library(dplyr)
library(purrr)

df <- df %>%
  mutate(
    size = accumulate(
      .x = 1 + growth,
      .f = ~ .x * .y,
      .init = first(size)
    )[-1]
  )

逻辑说明

• .x传入每年的(1+增长率)序列
• .f指定迭代规则：上一步的size结果(.x)乘以当前年的(1+增长率)(.y)
• .init指定迭代初始值为2018年的size
• 最后用[-1]去掉初始值生成的多余首位元素，匹配数据框行数

两种方法得到的结果和你原for循环的输出完全一致。

内容的提问来源于stack exchange，提问作者denisafonin