在R中如何基于长度动态变化的条件向量为数据框创建分组

问题原因 #

原for循环失效的核心问题有两个：

• 每次循环都会全量覆盖tour列，前序循环的赋值结果会被清空，仅保留最后一次循环的判定结果
• 循环边界设置错误：1:length(dates_tour)会导致最后一次循环取i+1时超出向量下标范围，触发索引报错

最优解决方案 #

方法1：使用findInterval（推荐，代码最简） #

该函数天然适配任意长度的断点向量，输出结果直接对应分组编号，完全匹配你原硬编码的区间逻辑：

library(dplyr)
# 先对日期断点排序，避免输入顺序异常导致错误
dates_tour_sorted <- sort(dates_tour)
df <- df %>%
  mutate(tour = findInterval(Date, dates_tour_sorted, rightmost.closed = TRUE))

方法2：使用cut函数 #

适合需要自定义分组标签的场景：

library(dplyr)
dates_tour_sorted <- sort(dates_tour)
df <- df %>%
  mutate(tour = as.integer(cut(Date,
                               breaks = dates_tour_sorted,
                               labels = seq_len(length(dates_tour_sorted)-1),
                               include.lowest = TRUE,
                               right = TRUE)))

效果验证 #

以上两种方法的输出结果和你原硬编码case_when生成的tour列完全一致，且无需修改代码即可自动适配任意长度的dates_tour断点向量。

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