La fonction switch()

Fait partie de la section Logique et flux d’exécution du Journey R de Coddy. Leçon 47 sur 64.

switch(x, name = value, ...) compare le texte x avec les noms et renvoie la valeur de celui qui correspond. Il remplace une longue chaîne de comparaisons else if portant sur la même variable :

size <- "m"
price <- switch(size, s = 2.5, m = 3, l = 3.5)
print(price)

Résultat :

[1] 3

Une dernière valeur sans nom est la valeur par défaut, utilisée lorsque rien ne correspond. Sans valeur par défaut, un switch() sans correspondance renvoie NULL de manière invisible :

day <- "fri"
print(switch(day, sat = "rest", sun = "rest", "work"))
print(is.null(switch("x", a = 1)))

Sortie :

[1] "work"
[1] TRUE

Une valeur vide passe à la suivante, de sorte que plusieurs noms peuvent partager un résultat :

kind <- function(day) switch(day, sat = , sun = "weekend", "weekday")
print(kind("sun"))
print(kind("tue"))

Sortie :

[1] "weekend"
[1] "weekday"

Les valeurs peuvent être des blocs entiers de code entre accolades, et le résultat est la dernière expression du bloc qui s’exécute :

area <- function(shape, a, b = a) {
  switch(shape,
    square = a^2,
    rect = a * b,
    circle = {
      r <- a / 2
      round(pi * r^2, 2)
    },
    NA)
}
print(area("rect", 3, 4))
print(area("circle", 10))

Sortie :

[1] 12
[1] 78.54

Défi

Facile

Complétez calculate(a, op, b). op est l’un de plus, minus, times et over. Retournez le résultat de son application à a et b, ou unknown operation pour toute autre valeur de op. Une division par zéro renvoie cannot divide by zero.

Le code fourni lit a, op et b (un par ligne) et affiche la valeur renvoyée.

Essayez vous-même

calculate <- function(a, op, b) {
  # Écrivez votre code ici
  "unknown operation"
}

# Code d'entrée/sortie fourni : conservez-le tel quel
input <- suppressWarnings(readLines(file("stdin")))
cat(calculate(as.numeric(input[1]), input[2], as.numeric(input[3])), sep = "\n")