在R语言中替换时序数据中特定序列的行值

问题说明 #

你需要处理包含Control、Fan、Compressor三类状态的时序数据，目标是将位于Control之后、Compressor之前的所有Fan状态替换为Compressor。现有代码仅能替换紧邻Compressor的最后一个Fan，无法批量替换中间的所有符合条件的Fan。

解决方案 #

核心思路是先定位「最近一次Control之后到首次出现Compressor之前」的区间，再批量替换该区间内的Fan。以下提供两种可行实现：

方法1：Base R 实现 #

# 复制原状态列，避免直接修改原始数据
df$Status_new <- df$Status

# 获取所有Control和Compressor的行号
control_rows <- which(df$Status == "Control")
compressor_rows <- which(df$Status == "Compressor")

# 遍历每个Control，处理其到后续第一个Compressor的区间
for (ctrl_row in control_rows) {
  # 找到当前Control之后的第一个Compressor行号
  next_compressor <- min(compressor_rows[compressor_rows > ctrl_row])
  # 若存在这样的Compressor，替换区间内的Fan
  if (length(next_compressor) > 0) {
    # 筛选区间内的Fan行号
    fan_to_replace <- which(df$Status[ctrl_row:next_compressor] == "Fan") + ctrl_row - 1
    df$Status_new[fan_to_replace] <- "Compressor"
  }
}

方法2：Tidyverse（dplyr）实现 #

利用窗口函数标记区间，代码更简洁：

library(dplyr)

df <- df %>%
  mutate(
    # 记录每个位置之前最近一次Control的行号
    last_control = cummax(ifelse(Status == "Control", row_number(), 0)),
    # 记录每个位置之后第一个Compressor的行号
    first_compressor = case_when(
      Status == "Compressor" ~ row_number(),
      TRUE ~ lead(row_number(), default = nrow(.))[cummax(ifelse(Status == "Compressor", row_number(), 0))]
    ),
    # 替换符合条件的Fan：在Control之后、Compressor之前的Fan
    Status_new = ifelse(
      Status == "Fan" & row_number() > last_control & row_number() < first_compressor,
      "Compressor",
      Status
    )
  ) %>%
  # 移除临时辅助列（可根据需要保留）
  select(-last_control, -first_compressor)

结果验证 #

用你提供的示例数据测试，两种方法都会生成期望的输出：

原代码问题分析 #

你的代码仅检查当前Fan的下一行是否为Compressor，因此只能替换紧邻Compressor的最后一个Fan。而上述方法先定位完整的Control到Compressor区间，再批量替换区间内所有Fan，覆盖了所有符合条件的位置。

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