R語言中如何編寫自己的函數初步入門

一、迴圈與控制

迴圈：

for(i in 1:10) print("hello world")

i<-10

while(i>0){

print(i);

i<-i-1;

}

控制

if（）

if（） else

ifelse（判斷，true，false）

switch（type，。。。）

二、使用者自訂函數

mystats <- function(x, parametric=TRUE, print=FALSE) {
  if (parametric) {
    center <- mean(x); spread <- sd(x) 
  } else {
    center <- median(x); spread <- mad(x) 
  }
  if (print & parametric) {
    cat("Mean=", center, "\n", "SD=", spread, "\n")
  } else if (print & !parametric) {
    cat("Median=", center, "\n", "MAD=", spread, "\n")
  }
  result <- list(center=center, spread=spread)
  return(result)
}

如何調用我們自己編寫的函數

set.seed(1234)
x <- rnorm(500) 產生符合常態分佈500個元素
y <- mystats(x)
y <- mystats(x, parametric=FALSE, print=TRUE)

下面是一個關於switch函數的例子

mydate <- function(type="long") {
    switch(type,
    long =  format(Sys.time(), "%A %B %d %Y"), 
    short = format(Sys.time(), "%m-%d-%y"),
    cat(type, "is not a recognized type\n"))
}
mydate("long")
mydate("short")
mydate()
mydate("medium")