R語(yǔ)言自定義函數(shù)

R語(yǔ)言某種程度來(lái)說(shuō)本身就是一個(gè)函數(shù)庫(kù),因?yàn)樗写罅康暮瘮?shù)可供調(diào)用,加上函數(shù)式編程,使得R語(yǔ)言的功能很強(qiáng)大。但是,有時(shí)候,根據(jù)實(shí)際需要,我們還是需要自己動(dòng)手編寫函數(shù),從而減少代碼的綴余與工作量。

與其它語(yǔ)言一樣,函數(shù)的組成莫過(guò)于關(guān)鍵字function、形參、是否返回值。函數(shù)的具體用法這里用幾個(gè)例子說(shuō)明

調(diào)用函數(shù)
這里寫一個(gè)打印函數(shù)演示簡(jiǎn)單的參數(shù)傳遞功能:

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

寫一個(gè)函數(shù)實(shí)現(xiàn)矩陣的乘法:

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)
}

我們構(gòu)造兩個(gè)矩陣

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

顯然x的列與y的行數(shù)不一樣,會(huì)不滿足矩陣的乘法。

> 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

函數(shù)的嵌套
在矩陣運(yùn)算函數(shù)中定義了一個(gè)打印新矩陣最大值函數(shù)

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語(yǔ)言函數(shù)與其它語(yǔ)言相比很簡(jiǎn)單、方便,實(shí)戰(zhàn)可能很復(fù)雜,需要根據(jù)實(shí)際情況,靈活運(yùn)用。





作者:柯廣的網(wǎng)絡(luò)日志

微信公眾號(hào):Java大數(shù)據(jù)與數(shù)據(jù)倉(cāng)庫(kù)