如何在R语言中按指定条件替换向量dr中的值？

给定向量dr，需按以下规则处理：

• 统计向量中大于0的元素个数
• 若个数超过5，保留原向量
• 若个数≤5，将向量所有元素替换为0

简洁实现代码 #

直接用条件判断即可完成，无需额外函数（需复用逻辑可封装成函数）：

# 直接处理向量dr
dr <- if (sum(dr > 0) > 5) dr else rep(0, length(dr))

如果需要重复使用该逻辑，封装成函数更方便：

process_vector <- function(vec) {
  pos_count <- sum(vec > 0)
  if (pos_count > 5) vec else rep(0, length(vec))
}

示例验证 #

示例1：保留原向量的情况

输入向量：

dr <- c(1,5,3,2,0,2,0,2,0,0,0,2,0)

处理后结果：

# 直接处理的结果
dr
# [1] 1 5 3 2 0 2 0 2 0 0 0 2 0

# 用函数处理的结果
process_vector(dr)
# [1] 1 5 3 2 0 2 0 2 0 0 0 2 0

示例2：替换为全0的情况

输入向量：

dr <- c(1,0,0,0,0,2,0,0,0,0,0,2,0)

处理后结果：

# 直接处理的结果
dr
# [1] 0 0 0 0 0 0 0 0 0 0 0 0 0

# 用函数处理的结果
process_vector(dr)
# [1] 0 0 0 0 0 0 0 0 0 0 0 0 0

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