R语言自编函数学习

函数向量化
自定义的函数，如果其中的计算都是向量化的， 那么函数自动地可以接受向量作为输入，结果输出向量。 比如，将每个元素都变成原来的平方的函数

f <- function(x){
  x^2
}

若为分段函数，则程序写为

`g <- function(x){
  if(abs(x) <= 1) {
    y <- x^2
  } else {
    y <- 1
  }
  
  y
}`

但是这个函数不能处理向量输入，因为if语句的条件必须是标量条件。 一个容易想到的修改是

gv <- function(x){
  y <- numeric(length(x))
  sele <- abs(x) <= 1
  y[sele] <- x[sele]^2
  y[!sele] <- 1.0
  
  y
}

或者

gv <- function(x){
  ifelse(abs(x) <= 1, x^2, 1)
}

或者

gv <- function(x){
  y <- numeric(length(x))
  for(i in seq(along=x)){
    if(abs(x[i]) <= 1) {
      y[i] <- x[i]^2
    } else {
      y[i] <- 1
    }
  }
  
  y
}

函数Vectorize可以将这样的操作自动化。如

g <- function(x){
  if(abs(x) <= 1) {
    y <- x^2
  } else {
    y <- 1
  }
  
  y
}
gv <- Vectorize(g)
gv(c(-2, -0.5, 0, 0.5, 1, 1.5))

来源：R语言教程，李东风