如何将含循环与ifelse的R函数输出以数值格式保存至R对象

核心问题总结 #

1. 无输出/无法保存结果：最初版本仅在循环内计算值，但未收集结果并返回；添加print仅输出到控制台，未将结果存入可返回的对象。
2. 非整数序列失效：矩阵版本错误地用x的元素值作为结果矩阵的索引，而非元素的位置下标。非整数无法作为合法索引，导致赋值混乱或报错。

推荐解决方案：向量化实现（R原生风格，高效简洁） #

ifelse是向量化函数，可直接处理整个x向量，无需循环，代码更简洁且避免索引错误：

fnct <- function(x, y) {
  # 预定义所有条件
  within_range <- x >= y[1] & x <= y[2]
  below_low <- x < y[1]
  global_cond <- max(x) - y[2] > y[1]
  
  # 向量化计算结果
  res <- ifelse(within_range, y[3],
                ifelse(below_low, y[1],
                       ifelse(global_cond, y[2], NA)))
  return(res)
}

测试验证

• 整数序列测试：

int_result <- fnct(c(1:150), c(20, 80, 50))
head(int_result) # 输出：20 20 20 20 20 20
tail(int_result) # 输出：80 80 80 80 80 80

• 非整数序列测试：

nonint_result <- fnct(seq(1.575, 5.125, by=0.05), c(1.5, 3, 3.5))
head(nonint_result) # 输出：3.5 3.5 3.5 3.5 3.5 3.5
tail(nonint_result) # 输出：3 3 3 3 3 3

可选：修正循环版本（若需保留循环逻辑） #

如果必须使用循环，需改用位置下标遍历x，而非直接遍历元素值：

fnct_loop <- function(x, y) {
  res <- numeric(length(x)) # 初始化数值向量存储结果
  max_x <- max(x)
  global_cond <- max_x - y[2] > y[1] # 提前计算全局条件，避免重复计算
  
  for (idx in seq_along(x)) {
    val <- x[idx]
    if (val >= y[1] & val <= y[2]) {
      res[idx] <- y[3]
    } else if (val < y[1]) {
      res[idx] <- y[1]
    } else {
      res[idx] <- if (global_cond) y[2] else NA
    }
  }
  return(res)
}

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