如何编写R函数实现将向量中大于等于指定值的元素替换为NA

你可以通过R内置的逻辑索引特性快速实现该功能，完整函数代码如下：

my_function <- function(x, y) {
  x[x >= y] <- NA
  return(x)
}

实现逻辑说明 #

• 利用R的向量运算特性，x >= y会生成和向量x长度一致的布尔值序列，对应位置元素满足条件时为TRUE，否则为FALSE
• 通过逻辑索引直接定位到符合条件的元素位置，批量赋值为NA，不需要循环遍历，执行效率更高

效果测试 #

调用你给出的示例参数验证：

my_function(x = c(1,2,3,4,5,6), y = 4)

输出结果为：

[1]  1  2  3 NA NA NA

和预期返回结果完全一致。

内容的提问来源于stack exchange，提问作者Jam.Wil