R语言自定义函数

R语言某种程度来说本身就是一个函数库,因为它有大量的函数可供调用,加上函数式编程,使得R语言的功能很强大。但是,有时候,根据实际需要,我们还是需要自己动手编写函数,从而减少代码的缀余与工作量。

与其它语言一样,函数的组成莫过于关键字function、形参、是否返回值。函数的具体用法这里用几个例子说明

调用函数
这里写一个打印函数演示简单的参数传递功能:

> a <-function(x){print(x)}
> a('hello world!')
[1] "hello world!"

写一个函数实现矩阵的乘法:

rm(list = ls())
a <-function(x,y)
{
  m1 <- ncol(x)
  n <- nrow(y)
  if(m1!=n)
  {
    print('error dimension is not siutable')
    return(0)
  }
  m <- nrow(x)
  n1 <- ncol(y)
  s <-matrix(0,m,n1)
  for(i in 1:m)
    for(j in 1:n1)
      s[i,j] <- sum(x[i,]*y[,j])
  return(s)
}

我们构造两个矩阵

> x <-matrix(c(1:6),2,3,byrow = TRUE)
> y <-matrix(c(1:8),2,4,byrow = TRUE)

显然x的列与y的行数不一样,会不满足矩阵的乘法。

> mat <- a(x,y)
[1] "error dimension is not siutable"
x <-matrix(c(1:6),2,3,byrow = TRUE)
y <-matrix(c(1:6),3,2,byrow = TRUE)x <-matrix(c(1:6),2,3,byrow = TRUE)
y <-matrix(c(1:6),3,2,byrow = TRUE)
mat <- a(x,y)
> mat
     [,1] [,2]
[1,]   22   28
[2,]   49   64

函数的嵌套
在矩阵运算函数中定义了一个打印新矩阵最大值函数

rm(list = ls())
a <-function(x,y)
{
  maxer <- function(x.)
  {
    print(max(x.))
  }
  m1 <- ncol(x)
  n <- nrow(y)
  if(m1!=n)
  {
    print('error dimension is not siutable')
    return(0)
  }
  m <- nrow(x)
  n1 <- ncol(y)
  s <-matrix(0,m,n1)
  for(i in 1:m)
    for(j in 1:n1)
      s[i,j] <- sum(x[i,]*y[,j])
  maxer(s)
  return(s)
}
x <-matrix(c(1:6),2,3,byrow = TRUE)
y <-matrix(c(1:6),3,2,byrow = TRUE)
mat <- a(x,y)
[1] 64

R语言函数与其它语言相比很简单、方便,实战可能很复杂,需要根据实际情况,灵活运用。





作者:柯广的网络日志

微信公众号:Java大数据与数据仓库