R语言如何将数据框列末尾最后一段连续1替换为0

需求说明 #

现有一个所有列值仅由0、1构成的数据框，需逐列处理，保证其余位置数值完全不变的前提下，仅将每列最靠后的一段连续1全部替换为0，示例如下：

# 原始输入数据
a <- c(0,1,1,0,1,1,1)
b <- c(0,1,1,1,0,1,1)
c <- data.frame(cbind(a,b))
head(c,7)

# 预期输出结果
a_desired <- c(0,1,1,0,0,0,0)
b_desired <- c(0,1,1,1,0,0,0)
c_desired <- data.frame(cbind(a_desired,b_desired))
head(c_desired,7)

之前尝试用tail()函数实现未找到可行方案。

实现代码 #

核心逻辑：逐列从末尾向前定位第一个出现0的位置，该位置之后的所有元素即为末尾连续1段，直接将这部分赋值为0即可，自动兼容全1列、末尾为0无后置1等边界场景。

# 单列处理函数
replace_last_continuous_ones <- function(x) {
  # 定位列中最靠近末尾的0的索引，无0则默认取0
  last_zero_idx <- max(which(x == 0), 0)
  # 将最后一个0之后的所有值置为0
  x[(last_zero_idx + 1):length(x)] <- 0
  return(x)
}

# 对数据框所有列批量应用处理函数
c_processed <- as.data.frame(lapply(c, replace_last_continuous_ones))

# 校验结果是否符合预期
all.equal(c_processed, c_desired) # 运行返回TRUE即为处理正确

边界场景适配说明 #

• 若列末尾本身为0、不存在后置连续1：此时最后一个0的索引等于列长度，后续无元素需要赋值，原列保持不变
• 若列全为1、无0值：最后一个0的索引取默认值0，会将整列所有元素置为0，符合处理逻辑
• 若列全为0：无1值需要替换，原列保持不变

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