优化R语言判断前后双值峰值的if/else函数，无需预处理数据

问题概述 #

现有函数需预处理添加零避免越界报错，且无法处理首尾元素，要求优化为无需预处理、首尾元素返回NULL，中间元素判断是否大于其前两个和后两个元素（返回逻辑值TRUE/FALSE）。

优化后的代码 #

judge_peaks <- function(y1) {
  # 初始化结果列表，长度与输入向量一致，默认元素为NULL
  y2 <- vector("list", length(y1))
  
  # 仅处理存在前后两个元素的位置：索引3到倒数第3位
  for (i in 3:(length(y1) - 2)) {
    y2[[i]] <- y1[i] > y1[i-1] && y1[i] > y1[i-2] && y1[i] > y1[i+1] && y1[i] > y1[i+2]
  }
  
  # 转换为向量，保留首尾的NULL值
  y2 <- unlist(y2)
  return(y2)
}

代码说明 #

• 初始化处理：用vector("list", length(y1))创建与输入等长的列表，转换为向量后首尾自然保留NULL，无需额外操作。
• 循环范围控制：将循环索引限制在3:(length(y1)-2)，确保i-2和i+2都不会超出向量边界，彻底避免越界报错，无需预处理添加零。
• 逻辑值返回：直接返回逻辑类型的TRUE/FALSE，而非字符串，符合R语言的数据类型规范，方便后续数据分析。

测试示例 #

# 测试向量
test_y1 <- c(2, 4, 7, 5, 3, 8, 10, 6, 2, 1)
# 调用函数
result <- judge_peaks(test_y1)
# 输出结果
print(result)

输出结果：

NULL NULL  TRUE FALSE FALSE FALSE  TRUE FALSE NULL NULL

原函数问题修正点 #

1. 原函数使用字符串'TRUE'/'FALSE'，改为逻辑值更贴合R的使用场景。
2. 原函数循环范围导致i+2越界，优化后通过限制循环区间解决，无需预处理数据。
3. 首尾元素通过初始化列表并转换为向量的方式自动返回NULL，无需单独处理。

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