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語言函數與其它語言相比很簡單、方便,實戰(zhàn)可能很復雜,需要根據實際情況,靈活運用。
作者:柯廣的網絡日志
微信公眾號:Java大數據與數據倉庫