在R语言中替换数据框列中相同高值间的1为对应高值

针对你需求中「将两个相同的非0非1数字之间的1替换为该数字，保留0和单独1」的场景，提供两种可行的R语言实现方案：

方法一：使用dplyr（通用兼容型） #

这种方法无需预设高值成对出现，能自动处理各种边界情况：

library(dplyr)

# 原始数据
data <- data.frame(col = c(0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0, 2, 1, 1, 1, 1, 2, 0, 0, 0, 0, 3, 1, 1, 3, 0, 0, 0, 0, 4, 1, 1, 1, 1, 4, 0, 0, 0, 5, 1, 1, 1, 1, 1, 5, 0, 0, 0))
# 预期结果（用于验证）
desired.outut.data <- data.frame(col = c(0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0, 2, 2, 2, 2, 2, 2, 0, 0, 0, 0, 3, 3, 3, 3, 0, 0, 0, 0, 4, 4, 4, 4, 4, 4, 0, 0, 0, 5, 5, 5, 5, 5, 5, 5, 0, 0, 0))

# 处理逻辑
result <- data %>%
  # 生成分组ID：每个以>1数字开头的连续段为一组
  mutate(group_id = cumsum(col > 1 & lag(col, default = -1) != col)) %>%
  group_by(group_id) %>%
  mutate(
    col = ifelse(
      # 判断当前组是否是「相同高值包裹1」的结构
      n() >= 2 && first(col) == last(col) && first(col) > 1,
      first(col),  # 替换为该高值
      col          # 其他情况保留原值
    )
  ) %>%
  ungroup() %>%
  select(-group_id)

# 验证结果是否符合预期
all.equal(result$col, desired.outut.data$col)  # 返回TRUE即正确

逻辑说明 #

1. group_id 会把每一段「高值+中间1+同高值」的连续序列单独分组
2. 对每个分组，检查首尾是否为同一个大于1的数字，若是则将整组替换为该数字
3. 单独的1、0以及单个出现的高值都会被原样保留

方法二：Base R（简洁高效型） #

如果可以确定数据中所有大于1的数字都是成对出现的，这种方法更直接高效：

# 原始数据
data <- data.frame(col = c(0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0, 2, 1, 1, 1, 1, 2, 0, 0, 0, 0, 3, 1, 1, 3, 0, 0, 0, 0, 4, 1, 1, 1, 1, 4, 0, 0, 0, 5, 1, 1, 1, 1, 1, 5, 0, 0, 0))
desired.outut.data <- data.frame(col = c(0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0, 2, 2, 2, 2, 2, 2, 0, 0, 0, 0, 3, 3, 3, 3, 0, 0, 0, 0, 4, 4, 4, 4, 4, 4, 0, 0, 0, 5, 5, 5, 5, 5, 5, 5, 0, 0, 0))

# 定位所有大于1的数字的位置
high_val_positions <- which(data$col > 1)

# 遍历每一对高值（步长为2，因为成对出现）
for(i in seq(1, length(high_val_positions), 2)){
  if(i + 1 > length(high_val_positions)) break
  start <- high_val_positions[i]
  end <- high_val_positions[i + 1]
  # 如果首尾高值相同，替换区间内所有值
  if(data$col[start] == data$col[end]){
    data$col[start:end] <- data$col[start]
  }
}

# 验证结果
all.equal(data$col, desired.outut.data$col)  # 返回TRUE即正确

逻辑说明 #

1. 先找出所有大于1的数字的索引位置
2. 按步长2遍历这些位置，每一对对应一个「高值-1-高值」的区间
3. 若这对高值相同，直接把区间内的所有值替换为该高值

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