ifelse() vectorisé

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

if teste une condition. Avec un vecteur contenant plusieurs valeurs, il s’arrête avec l’erreur the condition has length > 1. ifelse(test, yes, no) fonctionne élément par élément et renvoie un résultat par élément :

temps <- c(12, 25, 31, 18)
print(ifelse(temps > 20, "warm", "cool"))

Résultat :

[1] "cool" "warm" "warm" "cool"

yes et no peuvent être des vecteurs de même longueur, ainsi chaque élément peut obtenir sa propre valeur calculée :

price <- c(40, 120, 75)
final <- ifelse(price > 100, price * 0.9, price)
print(final)

Sortie :

[1]  40 108  75

Imbriquez les appels à ifelse() pour obtenir plus de deux résultats. Les conditions sont vérifiées dans l’ordre pour chaque élément :

scores <- c(95, 72, 48, 81)
grade <- ifelse(scores >= 90, "A", ifelse(scores >= 70, "B", "C"))
print(grade)

Sortie :

[1] "A" "B" "C" "B"

ifelse() est utile pour une nouvelle colonne d’une trame de données, où chaque ligne a besoin de sa propre étiquette :

df <- data.frame(item = c("tea", "oil"), stock = c(0, 8))
df$status <- ifelse(df$stock == 0, "sold out", "available")
print(df)

Résultat :

  item stock    status
1  tea     0  sold out
2  oil     8 available

Défi

Facile

Complétez shipping(weights). Renvoyez le coût d’expédition de chaque colis, relié par , : les colis jusqu’à 1 kg coûtent 4, jusqu’à 5 kg coûtent 9, et les colis plus lourds coûtent 2 par kg. Utilisez ifelse(), pas une boucle.

Le code fourni lit les poids sous forme de ligne séparée par des virgules et affiche la valeur renvoyée.

Essayez vous-même

shipping <- function(weights) {
  # Écrivez votre code ici
  ""
}

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