R：绘制分段函数

Question

我在R中定义的分段函数存在一些问题。

我将函数定义为：

g1 <- function(x) {
  if (x <= 0.25) {
    y <- gs1(x)
  }
  else if (x >= 0.75) {
    y <- gs3(x)
  }
  else {y <- gs2(x)}
  y
}

其中gs1，gs2，gs3是我之前定义的函数。

为了绘制函数g1，我尝试：

curve(g1)

但是R显示以下内容：

Warning message:
In if (x <= 0.25) { :
  the condition has length > 1 and only the first element will be used

我认为问题可能是R要求函数的参数为​​数字而不是向量？ 我不确定我是否正确。

我通过使用以下方法找出了绘图问题：

xx <- seq(0,1,length=200)
yy <- apply(as.matrix(xx),1,g1)
plot(xx,yy,type='l')

但是我仍然想问有什么方法可以解决这种问题，因为我发现我定义的分段函数对于许多其他命令也不可行。 例如，假设我想将g1集成到0到1之间，如果我只是尝试在R中“集成”功能，则会出现相同的警告消息。

Answer 1

您的问题是您正在寻找向量化if-else，尝试ifelse或使用result<-sapply(vector, g1)应用函数

Answer 2

您需要定义函数，以便它可以直接接受向量作为输入。

一种方法是

g1 <- function(x) 
  (x <= 0.25)*gs1(x) + (x <= 0.75 & x > 0.25)*gs3(x) + (x > 0.75)*gs2(x)

或者，

g1 <- function(x) {
  y <- gs1(x)
  y[x > 0.25] <- gs3(x[x > 0.25])
  y[x > 0.75] <- gs2(x[x > 0.75])
  y
}