Comment écrire une instruction de cas dans R (avec exemple)

Une instruction case est un type d’instruction qui passe en revue des conditions et renvoie une valeur lorsque la première condition est remplie.

Le moyen le plus simple d’implémenter une instruction case dans R consiste à utiliser la fonction case_when() du package dplyr :

library(dplyr)

df %>% 
  mutate(new_column = case_when(
    col1 < 9 ~ 'value1',
    col1 < 12 ~ 'value2',
    col1 < 15 ~ 'value3',
    TRUE ~ 'Great'))

Cette fonction particulière examine la valeur dans la colonne appelée col1 et renvoie :

• « value1 » si la valeur dans col1 est inférieure à 9
• « value2 » si la valeur dans col1 est inférieure à 12
• « value3 » si la valeur dans col2 est inférieure à 15
• « valeur4 » si aucune des conditions précédentes n’est vraie

Notez que TRUE est équivalent à une instruction « else ».

L’exemple suivant montre comment utiliser cette fonction dans la pratique.

Exemple : instruction de cas dans R

Supposons que nous ayons le bloc de données suivant dans R :

#create data frame
df <- data.frame(player=c(1, 2, 3, 4, 5, 6, 7, 8, 9, 10),
                 points=c(6, 8, 9, 9, 12, 14, 15, 17, 19, 22))

#view data frame
df

   player points
1       1      6
2       2      8
3       3      9
4       4      9
5       5     12
6       6     14
7       7     15
8       8     17
9       9     19
10     10     22

Nous pouvons utiliser la syntaxe suivante pour écrire une instruction case qui crée une nouvelle colonne appelée classe dont les valeurs sont déterminées par les valeurs de la colonne points :

library(dplyr) 

#create new column using case statement
df %>% 
  mutate(class = case_when(
    points < 9 ~ 'Bad',
    points < 12 ~ 'OK',
    points < 15 ~ 'Good',
    TRUE ~ 'Great'))

   player points class
1       1      6   Bad
2       2      8   Bad
3       3      9    OK
4       4      9    OK
5       5     12  Good
6       6     14  Good
7       7     15 Great
8       8     17 Great
9       9     19 Great
10     10     22 Great

L’instruction case a examiné la valeur dans la colonne des points et a renvoyé :

• « Mauvais » si la valeur dans la colonne des points était inférieure à 9
• « OK » si la valeur dans la colonne des points était inférieure à 12
• « Bon » si la valeur dans la colonne des points était inférieure à 15
• « Génial » si aucune des conditions précédentes n’est vraie

La nouvelle colonne s’appelle class , puisque c’est le nom que nous avons spécifié dans la fonction mutate() .

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres tâches courantes dans R :

Comment utiliser l’instruction If avec plusieurs conditions dans R
Comment écrire une instruction If Else imbriquée dans R
Comment écrire votre première fonction tryCatch() dans R