R中data.frame如何实现仅数值达到触发阈值时分组编号递增

有一个按ID升序排序的data.frame，包含value数值列，需要实现如下分组逻辑：仅当value列出现>=指定阈值的数值时，分组编号才递增，初始分组编号为1。

先对每一行做是否满足触发条件的布尔判断（value >= 阈值），对生成的布尔向量做累加求和，最终结果加1即可得到符合要求的分组编号。

基础R实现 #

# 定义触发阈值
threshold <- 300
# 生成分组列
dat$group <- cumsum(dat$value >= threshold) + 1

dplyr（tidyverse生态）实现 #

library(dplyr)

threshold <- 300
dat <- dat %>%
  mutate(group = cumsum(value >= threshold) + 1)

data.table实现（适合大数据量场景） #

library(data.table)

threshold <- 300
setDT(dat)
dat[, group := cumsum(value >= threshold) + 1]

如果需要触发阈值的当前行仍归属上一分组，仅下一行开始分组编号递增，可以将布尔向量滞后一位处理：

# 基础R版
dat$group <- cumsum(c(FALSE, head(dat$value >= threshold, -1))) + 1

执行分组计算前必须确认数据已经按照ID完成升序排序，否则分组结果会不符合预期。

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