R语言：优化简单物质消耗相互作用模型的实现

我编写了一个简单数值示例（后续将拓展为更复杂版本），模拟两种物质A和B在每一轮过程迭代中以速率r相互消耗的过程。现有代码可正常运行，但实现不够优雅：一是依赖显式循环，二是存在存储初始状态且会被覆盖的变量。此外，模拟存在停止条件：物质A、B不能取负值，一旦出现负值需立即终止过程。以下是改进后的实现和说明：

改进后的代码 #

library(dplyr)
library(purrr)

# 定义模拟参数
initial_A <- 2000
initial_B <- 1000
consumption_rate <- 1/10
max_iterations <- 10

# 单轮消耗计算与终止条件检查函数
calculate_next_state <- function(current_state, rate) {
  A_current <- current_state$A
  B_current <- current_state$B
  
  # 计算下一轮物质剩余量
  A_next <- A_current - B_current * rate
  B_next <- B_current - A_current * rate
  
  # 检查是否触发终止条件（出现负值）
  if (A_next < 0 || B_next < 0) {
    return(tibble(A = max(A_next, 0), B = max(B_next, 0), stop_simulation = TRUE))
  }
  
  tibble(A = A_next, B = B_next, stop_simulation = FALSE)
}

# 初始化模拟状态
starting_state <- tibble(A = initial_A, B = initial_B, stop_simulation = FALSE)

# 执行迭代模拟
simulation_results <- accumulate(
  .init = starting_state,
  1:max_iterations,
  function(prev_state, .) {
    if (prev_state$stop_simulation) {
      prev_state # 触发终止后保持当前状态，不再继续迭代
    } else {
      calculate_next_state(prev_state, consumption_rate)
    }
  }
) %>%
  bind_rows() %>%
  # 截取到第一次触发终止的行（包含终止状态）
  slice(1:which(stop_simulation)[1]) %>%
  select(-stop_simulation) # 移除终止标记列

# 查看结果
simulation_results

关键改进说明 #

• 替换显式循环为purrr::accumulate：accumulate专门用于状态迭代场景，自动维护每一轮的状态传递和结果收集，代码更简洁易读。
• 消除变量覆盖问题：初始状态仅作为迭代起点存储，后续每一轮状态都以独立的tibble形式保存，不存在变量被覆盖的情况。
• 内置终止条件检查：在单轮计算函数中直接判断是否出现负值，一旦触发就标记终止，避免无效迭代，同时保证结果符合非负要求。
• 结构化数据全程流转：从初始状态到每一轮结果都用tibble存储，无需后续再做格式转换、重命名等冗余操作，适配后续复杂拓展需求。

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