2019年8月8日 星期四

R語言的函數回傳

R語言中,函數回傳有3種方式,以下分別說明。
  1. 函數中有return。
  2. 函數中無return。
  3. 函數可多值回傳。
函數中有return
test1 <- function(x, y) {
  z <- x + y
  return(z)
}
test1(5, 3)
test1(x = 5, y = 3) 
test1(y = 3, x = 5) 
執行結果:
PS:回傳需要小括號。

函數中無return
test2 <- function(x, y, z) {
  r <- (x + y) * z
  r
}
test2(1, 5, 7)
test2(x = 1, y = 5, z = 7) 
test2(z = 7, x = 1, y = 5) 
執行結果:

PS:在R語言中,將會自動回傳函數最後一行的數值,即使不用寫 return,數值依舊可以回傳,但就閱讀而言,函數有寫 return 可讓程式碼方便閱讀。

函數可多值回傳
test3 <- function(x, y, z) {
  r1 <- (x + y) * z
  r2 <- (x - y) * z
  r3 <- (x * y) / z
  list(r1, r2, r3)
}
test3(2, 8, 5)
test3(x = 2, y = 8, z = 5) 
test3(z = 5, x = 2, y = 8) 
執行結果:

test3 <- function(x, y, z) {
  r1 <- (x + y) * z
  r2 <- (x - y) * z
  r3 <- (x * y) / z
  c(r1, r2, r3)
}
test3(2, 8, 5)
test3(x = 2, y = 8, z = 5) 
test3(z = 5, x = 2, y = 8) 
執行結果:
參考資料:


沒有留言:

張貼留言