使用R矢量化我的函数不起作用

时间:2019-10-19 11:32:41

标签: r function vectorization

我有这个功能:

fx <- function (x) {
  if (x <= 0) {
    return(-1 * (x**3))
  }
  else if (0 < x && x <= 1) {
    return(x**2)
  }
  else if (x > 1) {
    return(sqrt(x))
  }
}

我转换为向量化函数为:

fVectorize <- function(x) {
  result <- ifelse(x <= 0, -1 * (x**3), ifelse(0 < x && x <= 1, x**2, sqrt(x)))
  result
}

我尝试了以下值,但不起作用:

fVectorize(c(-2, -4, -5))
fVectorize(c(0.5, 0.3, 0.7))
fVectorize(c(3, -4, 0.7))

好像只是在第一个条件。

有什么主意吗?

1 个答案:

答案 0 :(得分:5)

&&更改为&

fVectorize <- function(x) {
  result <- ifelse(x <= 0, -1 * (x**3), ifelse(0 < x & x <= 1, x**2, sqrt(x)))
  result
}
fVectorize(c(-2, -4, -5))
#[1]   8  64 125

fVectorize(c(0.5, 0.3, 0.7))
#[1] 0.25 0.09 0.49

fVectorize(c(3, -4, 0.7))
# [1]  1.732051 64.000000  0.490000

&进行元素明智的操作,其中&&看起来是第一个返回值的元素。