Comment ajouter une colonne si elle n’existe pas dans R

Vous pouvez utiliser la fonction personnalisée suivante pour ajouter une ou plusieurs colonnes à un bloc de données dans R si elles n’existent pas déjà :

add_cols <- function(df, cols) {
  add <- cols[!cols %in% names(df)]
  if(length(add) != 0) df[add] <- NA
  return(df)
}

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

Exemple : ajouter une colonne si elle n’existe pas dans R

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

#create data frame
df <- data.frame(team=c('A', 'A', 'A', 'A', 'B', 'B', 'B'),
                 position=c('Gu', 'Fo', 'Fo', 'Fo', 'Gu', 'Gu', 'Fo'),
                 points=c(18, 22, 19, 14, 14, 11, 20))

#view data frame
df

  team position points
1    A       Gu     18
2    A       Fo     22
3    A       Fo     19
4    A       Fo     14
5    B       Gu     14
6    B       Gu     11
7    B       Fo     20

Supposons que nous souhaitions ajouter les colonnes suivantes au bloc de données si elles n’existent pas déjà :

• points
• aide
• rebondit

Nous pouvons utiliser une fonction personnalisée appelée add_cols pour ce faire :

#define custom function to add columns to data frame if they do not exist
add_cols <- function(df, cols) {
  add <- cols[!cols %in% names(df)]
  if(length(add) !=0 ) df[add] <- NA
  return(df)
}

#add three columns if they don't already exist
df <- add_cols(df, c('points', 'assists', 'rebounds'))

#view updated data frame
df

  team position points assists rebounds
1    A       Gu     18      NA       NA
2    A       Fo     22      NA       NA
3    A       Fo     19      NA       NA
4    A       Fo     14      NA       NA
5    B       Gu     14      NA       NA
6    B       Gu     11      NA       NA
7    B       Fo     20      NA       NA

Notez que les colonnes de passes décisives et de rebonds ont été ajoutées au bloc de données alors que la colonne de points ne l’était pas puisqu’elle existait déjà.

Notez également que R remplit simplement chaque valeur dans les nouvelles colonnes avec des valeurs NA.

Ressources additionnelles

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

Comment ajouter une colonne au bloc de données en fonction d’autres colonnes dans R
Comment ajouter une colonne d’index (ID numérique) à un bloc de données dans R
Comment ajouter une colonne vide à un bloc de données dans R