R语言无需循环移除向量末尾特定NA值的最优实现方法

方案背景 #

需求为仅移除向量末尾的连续NA值，保留向量开头、中间的NA值，示例向量如下：
vec <- c(1:5, NA, 6:10, NA, NA, NA)
预期处理结果：

> vec
[1]  1  2  3  4  5 NA  6  7  8  9 10

原本使用的倒序while循环方案存在逻辑写法不严谨的问题（is.na(vec[i]==TRUE)的判断顺序错误，正确写法应为is.na(vec[i])），且循环在处理长向量时性能较差。

无循环最优方案 #

方案1：向量化cummax方案（性能最优，适合长向量） #

核心逻辑是利用倒序累积最大值标记末尾NA位置，全程无循环，是所有方案中性能最高的实现：

# 一行代码实现
vec <- vec[rev(cummax(rev(!is.na(vec)))) == 1]

逻辑拆解：

1. !is.na(vec)：判断每个位置是否为非NA值，返回逻辑向量
2. 第一次rev()：将逻辑向量倒序
3. cummax()：计算累积最大值，倒序后第一个非NA值出现的位置之后所有值都会被标记为1
4. 第二次rev()：将逻辑序列恢复原顺序，此时所有末尾NA的位置会被标记为0，直接过滤即可

方案2：定位最后一个非NA值截断（可读性最高） #

如果更看重代码可读性，可使用Position函数直接定位最后一个非NA的位置，再截断向量：

last_non_na_pos <- Position(function(x) !is.na(x), vec, right = TRUE)
vec <- vec[seq_len(last_non_na_pos)]

效果验证 #

用示例向量测试两种方案，均可得到符合预期的结果：

vec <- c(1:5, NA, 6:10, NA, NA, NA)
# 用方案1测试
vec <- vec[rev(cummax(rev(!is.na(vec)))) == 1]
print(vec)
# 输出：[1]  1  2  3  4  5 NA  6  7  8  9 10

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