R语言ifelse实现分段函数结果错误，如何修正代码？

错误原因 #

你的函数中两个ifelse是独立执行的，第一个ifelse计算出的结果没有被保存或返回，函数最终仅返回第二个ifelse的输出。以x=-2为例，第二个ifelse判断x>1不成立，返回0，导致正确结果被覆盖。

修正方法 #

方法1：嵌套ifelse #

将多个条件嵌套在同一个ifelse结构中，确保所有分支的结果都能被正确返回：

pwfun = function(x) {
  ifelse(x < -1, -2*x - 2,
         ifelse(x > 1, x^2 - 1, 0))
}

测试调用：

pwfun(c(-2, 0.5, 3))
# 输出：[1] 2 0 8

方法2：分步向量赋值 #

初始化结果向量后，针对不同条件逐个赋值，逻辑更清晰，适合多分段场景：

pwfun = function(x) {
  res = numeric(length(x))  # 创建与输入长度一致的空结果向量
  res[x < -1] = -2 * x[x < -1] - 2
  res[x > 1] = x[x > 1]^2 - 1
  res
}

调用后同样会得到正确结果。

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