Functies schrijven

In R zijn functies eersteklas burgers (op zichzelf staande objecten) en stellen ze je in staat je code te structureren in modulaire, herbruikbare en goed georganiseerde blokken.

Een functie definiëren

Om een ​​functie te definiëren, gebruikt u het trefwoord function en wijst u het gemaakte object toe aan een variabele (de functienaam) met behulp van de toewijzingsoperator <-:

# Simple function definition
greet_user <- function() {
  print("Hello, User!")
}

# Invoking the function
greet_user()

Parameters en retourwaarden

Functies kunnen een of meer parameters accepteren. Als u expliciet een waarde wilt retourneren, gebruikt u de functie return(). Als return() wordt weggelaten, retourneert R automatisch de waarde van de laatst geëvalueerde expressie in de hoofdtekst van de functie.

# Function that adds two numbers and returns the result
add_numbers <- function(a, b) {
  sum_result <- a + b
  return(sum_result)
}

total <- add_numbers(5, 7)
print(total) # Prints 12

Standaardargumenten

U kunt standaardwaarden opgeven voor de parameters van een functie. Als het argument niet wordt opgegeven wanneer de functie wordt aangeroepen, gebruikt R de standaardwaarde:

# The function uses a default discount rate of 10% (0.10)
apply_discount <- function(price, discount = 0.10) {
  discounted_price <- price * (1 - discount)
  return(discounted_price)
}

# Uses the default (10%)
print(apply_discount(100)) # Prints 90

# Specifies a custom discount (20%)
print(apply_discount(100, 0.20)) # Prints 80

Meerdere waarden retourneren met lijsten

Omdat een functie in R slechts één object rechtstreeks via return() kan retourneren, kunnen we, als we meerdere waarden moeten retourneren (mogelijk van verschillende typen of lengtes), deze in een lijst bundelen (list):

get_stats <- function(v) {
  stats <- list(
    mean_val = mean(v),
    max_val = max(v)
  )
  return(stats)
}

results <- get_stats(c(10, 20, 30))
print(results$mean_val) # Prints 20

Anonieme functies en sapply

Anonieme functies (functies zonder inline gedefinieerde naam) zijn uiterst handig in combinatie met functies zoals sapply() om snelle transformaties toe te passen op elementen van een vector of lijst.

In moderne R (vanaf versie 4.1.0) kun je anonieme functies definiëren met behulp van de verkorte syntaxis \(x) naast de klassieke function(x):

v <- c(1, 2, 3)

# Using sapply with a classic anonymous function
sapply(v, function(x) x^2) # 1 4 9

# Using sapply with shorthand syntax (R >= 4.1.0)
sapply(v, \(x) x^2) # 1 4 9

Probeer het zelf