R语言：如何将分组后数据框指定列的最后N个值设为NA

要实现分组后将每组某列的最后N个值设为NA，且兼容任意N（包括分组大小小于等于N的情况），可以利用组内行号来判断需要保留的元素，避免索引越界的问题。

方法一：使用row_number()逻辑判断 #

这是最通用的写法，能处理所有边界情况（比如分组大小为1、分组大小小于N等）：

library(dplyr)

# 定义要设置为NA的最后N个元素数量
N <- 2

# 测试数据
df <- tibble(
  ID = c(1,1,1,2,2,3),
  target = c(10,20,30,40,50,60)
)

# 核心代码
df %>%
  group_by(ID) %>%
  mutate(target = ifelse(row_number() <= (n() - N), target, NA)) %>%
  ungroup()

逻辑说明：

• row_number()返回当前行在组内的序号（从1开始）
• n()是当前分组的总行数
• n() - N表示每组需要保留的前多少个元素：
  • 如果分组行数n() > N：保留前n()-N个元素，后面N个设为NA
  • 如果分组行数n() <= N：n()-N≤0，所有行的row_number()都大于这个值，整组的target都会被设为NA

方法二：使用replace()函数（需额外处理边界） #

如果更喜欢用位置替换的方式，需要先判断分组大小是否大于N，避免无效索引：

df %>%
  group_by(ID) %>%
  mutate(
    target = if (n() > N) {
      replace(target, (n() - N + 1):n(), NA)
    } else {
      NA_real_
    }
  ) %>%
  ungroup()

为什么你之前的代码失效？ #

1. 直接用target[1:(n()-1)]的写法，当分组大小为1时，1:(1-1)是1:0，属于无效索引，导致报错。
2. 用ifelse(n()==1, target, ...)的问题在于：ifelse是向量化函数，会将n()==1这个标量判断广播到组内每一行，但后面的c(target[1:(n()-1)], NA)是一个长度等于组大小的向量，两者逻辑不匹配，导致else分支无法正确生效。

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