R语言：用循环替代手动迭代计算变量X的预测值

我有变量X在2021Q1至2022Q3的实际季度数据，以及2022Q4至2025Q4的预测增长率growth_x，需要基于growth_x迭代计算2022Q4到2025Q4的X值。目前手动分阶段计算时遗漏了2025Q4，作为R循环编程新手，希望能编写函数实现自动计算，求帮助。

原代码 #

library(readxl)
library(dplyr)
library(lubridate)

# Quarterly Data
data <- data.frame(c("2021Q1","2021Q2","2021Q3","2021Q4",
                     "2022Q1","2022Q2","2022Q3","2022Q4",
                     "2023Q1","2023Q2","2023Q3","2023Q4",
                     "2024Q1","2024Q2","2024Q3","2024Q4",
                     "2025Q1","2025Q2","2025Q3","2025Q4"),
                  
                   # Variable X - Actuals upto 2022Q3           
                   c(804,511,479,462,
                     427,330,440,NA,
                     NA,NA,NA,NA,
                     NA,NA,NA,NA,
                     NA,NA,NA,NA),
                  
                   # Forecasted Growth rates of X from 2022Q4       
                    c(NA,NA,NA,NA,
                      NA,NA,NA,0.24,
                      0.49,0.65,0.25,0.71,
                      0.63,0.33,0.53,0.83,
                      0.87,0.19,0.99,0.16))  

# Renaming the columns
data<-data%>%rename(yrqtr=1,x=2,growth_x=3)

# Creating Date Variable
data<-data%>%mutate(year=substr(yrqtr,1,4),
                    qtr=substr(yrqtr,5,6),
                    mon=ifelse(qtr=="Q1",3,
                               ifelse(qtr=="Q2",6,
                                      ifelse(qtr=="Q3",9,12))),
                                date=make_date(year,mon,1))

# Computing Growth Rate from 2022Q3 to 2023Q3
Growth_2023_3<-data%>%mutate(forecast_x=(1+growth_x)*lag(x,4),
                                       x=ifelse(date>"2022-09-01",forecast_x,x))%>%select(-forecast_x)

# Computing Growth Rate from 2023Q3 to 2024Q3
Growth_2024_3<-Growth_2023_3%>%mutate(forecast_x=(1+growth_x)*lag(x,4),
                                          x=ifelse(date>"2023-09-01",forecast_x,x))%>%select(-forecast_x)

# Computing Growth Rate from 2024Q3 to 2025Q3
Growth_2025_3<-Growth_2024_3%>%mutate(forecast_x=(1+growth_x)*lag(x,4),
                             x=ifelse(date>"2024-09-01",forecast_x,x))%>%select(-forecast_x)

解决方案：用循环自动计算所有预测值 #

原代码手动分阶段处理容易遗漏季度，以下是用循环实现的通用函数，能自动填充所有缺失的X值（包括2025Q4）：

library(dplyr)
library(lubridate)

# 定义计算函数
compute_forecast_x <- function(input_data) {
  # 复制数据避免修改原始数据，确保x为数值型
  result <- input_data %>% mutate(x = as.numeric(x))
  
  # 获取需要计算的行索引：x为空且增长率存在的行
  forecast_rows <- which(is.na(result$x) & !is.na(result$growth_x))
  
  # 循环计算每个预测季度的X值
  for (i in forecast_rows) {
    # 上一年同季度的索引（往前推4行）
    prev_qtr_idx <- i - 4
    # 计算当期X值：上一年同期X * (1+当期增长率)
    result$x[i] <- result$x[prev_qtr_idx] * (1 + result$growth_x[i])
  }
  
  return(result)
}

# 调用函数计算完整数据
final_data <- compute_forecast_x(data)

# 查看结果（仅展示关键列）
final_data %>% select(yrqtr, x, growth_x) %>% print(n = 20)

代码说明 #

1. 函数封装：将计算逻辑封装成函数，方便复用和维护
2. 自动识别待计算行：通过which()筛选出需要填充的行，无需手动指定时间范围
3. 循环迭代计算：对每个待计算行，自动取上一年同季度的X值，结合增长率计算当期值，不会遗漏任何季度
4. 数据安全：复制原始数据进行计算，避免修改原始数据

运行后，final_data会包含2021Q1至2025Q4完整的X值，包括之前遗漏的2025Q4。

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