Pandas : supprimez une colonne spécifique lors de l’importation d’un fichier CSV

Vous pouvez utiliser la syntaxe de base suivante pour supprimer une colonne spécifique lors de l’importation d’un fichier CSV dans un DataFrame pandas :

df = pd.read_csv('basketball_data.csv', usecols=lambda x: x != 'rebounds')

Cet exemple particulier lira chaque colonne d’un fichier CSV appelé basketball_data.csv dans un DataFrame pandas, à l’exception de la colonne appelée rebonds .

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

Exemple : supprimer une colonne spécifique lors de l’importation d’un fichier CSV dans Pandas

Supposons que nous ayons le fichier CSV suivant appelé basketball_data.csv :

Nous pouvons utiliser la syntaxe suivante pour importer le fichier CSV dans pandas et supprimer la colonne appelée rebonds lors de l’importation :

import pandas as pd

#import all columns except 'rebounds' into DataFrame
df = pd.read_csv('basketball_data.csv', usecols=lambda x: x != 'rebounds')

#view resulting DataFrame
print(df)

  team  points
0    A      22
1    B      14
2    C      29
3    D      30

Notez que la colonne des rebonds a été supprimée lorsque nous avons importé le fichier CSV dans pandas.

Si vous souhaitez supprimer plusieurs colonnes lors de l’importation, vous pouvez utiliser l’opérateur not in comme suit :

import pandas as pd

#import all columns except 'team' and 'rebounds' into DataFrame
df=pd.read_csv('basketball_data.csv', usecols=lambda x: x not in ['team', 'rebounds'])

#view resulting DataFrame
print(df)

   points
0      22
1      14
2      29
3      30

Notez que les colonnes équipe et rebonds ont toutes deux été supprimées lorsque nous avons importé le fichier CSV dans pandas.

Notez que vous pouvez inclure autant de noms de colonnes que vous le souhaitez dans la liste suivant l’opérateur not in pour supprimer autant de colonnes que vous le souhaitez lors de l’importation d’un fichier CSV.

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres tâches courantes en Python :

Pandas : Comment ignorer des lignes lors de la lecture d’un fichier CSV
Pandas : Comment ajouter des données à un fichier CSV existant
Pandas : comment utiliser read_csv avec l’argument usecols