在dplyr的mutate中使用含for循环的函数是否可行？有无更优方案？

原方案的合理性与问题 #

你的核心思路是对的：通过追踪连续高温天数来标记热浪，但当前实现存在两个明显问题：

1. 效率瓶颈：days_in_current_heatwave = c(days_in_current_heatwave, i)每次循环都在动态扩容向量，大数据量下会显著拖慢速度——R里向量扩容需要重新分配内存，频繁操作的成本很高。
2. 逻辑不完整：伪代码里的... some more code部分缺失了关键逻辑：当连续高温天数达到3天及以上时，需要把这些天（包括之前的天数）都标记为热浪，而目前只在统计连续天数，没有完成最终的标记输出。

更优实现思路 #

1. 优化原生for循环（解决效率与逻辑问题） #

如果坚持用循环，提前分配好结果向量，并且在热浪中断或循环结束时统一标记符合条件的连续段，避免动态扩容：

is_heatwave <- function(vals) {
  n <- length(vals)
  day_in_heatwave <- logical(n)
  current_streak <- 0
  streak_start <- 1
  
  for (i in 1:n) {
    if (vals[i] > 30) {
      if (current_streak == 0) streak_start <- i
      current_streak <- current_streak + 1
    } else {
      # 热浪中断，检查是否满足连续3天条件
      if (current_streak >= 3) {
        day_in_heatwave[streak_start:(i-1)] <- TRUE
      }
      current_streak <- 0
    }
  }
  # 处理循环结束时仍在持续的热浪
  if (current_streak >= 3) {
    day_in_heatwave[streak_start:n] <- TRUE
  }
  
  return(day_in_heatwave)
}

2. 用rle游程编码（高效简洁） #

R的rle函数专门处理连续重复值的场景，代码更简洁，效率远高于循环，尤其适合大数据量：

is_heatwave_rle <- function(vals) {
  # 先标记单日是否高温
  hot_days <- vals > 30
  # 对高温标记做游程编码
  rle_result <- rle(hot_days)
  # 将长度≥3的连续高温段标记为热浪
  rle_result$values[rle_result$lengths >= 3] <- TRUE
  # 长度不足3的连续高温段标记为非热浪
  rle_result$values[rle_result$lengths < 3] <- FALSE
  # 还原为原长度的结果向量
  inverse.rle(rle_result)
}

3. 用slider包适配tidyverse流程 #

如果你习惯用dplyr的链式操作，slider包的滑动窗口函数可以无缝融入mutate，代码更符合tidyverse风格：

library(dplyr)
library(slider)

# 直接在mutate中实现，无需单独写函数
df %>%
  mutate(
    # 标记单日高温
    hot = temps > 30,
    # 计算当前位置的连续高温长度（从当前往前到最近的非高温天）
    streak = slide_dbl(hot, ~sum(.x), .before = Inf, .complete = FALSE),
    # 判断当前天是否属于连续≥3天的高温段（当前或未来2天的连续长度达标即可）
    is_heatwave = streak >=3 | lead(streak, 1, default=0)>=3 | lead(streak,2, default=0)>=3
  ) %>%
  select(-hot, -streak)

测试验证 #

用你的测试数据验证：

df = data.frame(
  temps = c(30,30,32,30,24)
)

# 所有天都不满足连续3天超30℃，结果全为FALSE
df %>% mutate(is_heatwave = is_heatwave_rle(temps))

如果测试数据是c(31,32,33,29,34,35,36)，结果会正确标记前3天和后3天为TRUE。

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