先前提到了R 中向量可以作为单独的参数传入函数中,函数会将参数中的元素一一执行并返回一个对应原先向量的新的向量:
> my_f = function(x){ x*3 }
> my_f(c(1,2,3,4,5))
[1] 3 6 9 12 15
但如果我们的函数中有条件句等接受标量的语句存在,就不能直接使用了:
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
}
- ifelse 语句
gv <- function(x){
ifelse(abs(x) <= 1, x^2, 1)
}
- Vectorize 直接创建新的向量化函数
> gv <- Vectorize(g)
> gv
function (x)
{
args <- lapply(as.list(match.call())[-1L], eval, parent.frame())
names <- if (is.null(names(args)))
character(length(args))
else names(args)
dovec <- names %in% vectorize.args
do.call("mapply", c(FUN = FUN, args[dovec], MoreArgs = list(args[!dovec]),
SIMPLIFY = SIMPLIFY, USE.NAMES = USE.NAMES))
}
<environment: 0x7facc2ab8d48>
这时候可以直接调用新函数gv 了:
gv(c(-2, -0.5, 0, 0.5, 1, 1.5))
## [1] 1.00 0.25 0.00 0.25 1.00 1.00
- sapply 函数
> sapply(c(-2, -0.5, 0, 0.5, 1, 1.5), g)
[1] 1.00 0.25 0.00 0.25 1.00 1.00