R语言dataframe单列NA按遇1重置规则替换为递增序列问题

• 变量x的"NA"是字符串类型，不是R原生缺失值NA，is.na()对字符串"NA"会返回FALSE，判断逻辑完全失效。
• 代码混用对象名df和df1，运行时会直接报错找不到对象。
• 核心逻辑不符合需求：seq(2,100)是全局连续序列，无法实现遇到值为1的行就重置计数的分段递增效果。

基础R实现 #

# 先将字符串"NA"转为真正的数值型缺失值
df$x <- as.numeric(ifelse(df$x == "NA", NA, df$x))
# 生成分组标签：每遇到1就新增一个分组
df$group <- cumsum(!is.na(df$x) & df$x == 1)
# 分组内生成符合要求的递增序列
df$Sequence <- ave(df$x, df$group, FUN = function(v) {
  # 首个分组（第一个1之前的行）从2开始计数，其余分组从1开始对应x=1的位置
  seq_len(length(v)) + ifelse(is.na(v[1]), 1, 0)
})

# 查看结果
print(df$Sequence)

dplyr实现（代码更简洁） #

library(dplyr)
df <- df %>%
  mutate(
    x = as.numeric(ifelse(x == "NA", NA, x)),
    group = cumsum(x == 1 & !is.na(x)),
    Sequence = row_number() + ifelse(is.na(first(x)), 1, 0),
    .by = group
  )

运行后输出的Sequence列完全符合需求：第一个1之前的缺失值从2开始递增，遇到1后后续缺失值重新从2开始计数，直到下一个1出现。

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