R语言中如何实现匹配指定值的行正向及反向连续计数

核心逻辑拆解 #

实现需求的核心分为两步：

• 拆分数据框内x1列的连续值段，为每一段独立分配组标识
• 针对组内值等于指定匹配值的片段，根据计数方向生成连续序号，非匹配值位置统一置0

基础R（Base R）实现 #

无需依赖额外包，直接基于内置函数完成：

# 构建用户提供的测试数据
df <- data.frame(x1 = c(rep(0, 3), rep(1, 4), rep(0, 4)))
# 设定需要匹配的目标值
target_val <- 1
# 基于游程编码拆分连续值段
rle_res <- rle(df$x1)
# 生成正向连续计数列
df$x2_forward <- unlist(lapply(seq_along(rle_res$values), function(idx) {
  seg_length <- rle_res$lengths[idx]
  if (rle_res$values[idx] == target_val) {
    seq_len(seg_length)
  } else {
    rep(0, seg_length)
  }
}))
# 生成反向连续计数列
df$x2_backward <- unlist(lapply(seq_along(rle_res$values), function(idx) {
  seg_length <- rle_res$lengths[idx]
  if (rle_res$values[idx] == target_val) {
    rev(seq_len(seg_length))
  } else {
    rep(0, seg_length)
  }
}))

Tidyverse生态实现 #

适配常用的dplyr操作流，代码可读性更强：

library(dplyr)
df <- data.frame(x1 = c(rep(0, 3), rep(1, 4), rep(0, 4)))
target_val <- 1
df <- df %>%
  mutate(
    x2_forward = ifelse(x1 == target_val, row_number(), 0),
    x2_backward = ifelse(x1 == target_val, max(row_number()) - row_number() + 1, 0),
    .by = consecutive_id(x1)
  )

注：consecutive_id()为dplyr 1.1.0及以上版本的内置函数，可自动识别连续相同值并完成分组，不需要手动编写差值判断逻辑。

输出效果说明 #

基于示例x1列的计算结果完全符合连续计数规则：

• 正向计数：从匹配值块的第一行开始向后累加，结果为 0, 0, 0, 1, 2, 3, 4, 0, 0, 0, 0
• 反向计数：从匹配值块的最后一行开始向前累加，结果为 0, 0, 0, 4, 3, 2, 1, 0, 0, 0, 0

修改target_val的赋值，即可适配任意指定匹配值的计数需求，无需调整其余逻辑。

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