Pandas : exporter des colonnes spécifiques dans DataFrame vers un fichier CSV

Vous pouvez utiliser la syntaxe suivante pour exporter uniquement des colonnes spécifiques d’un DataFrame pandas vers un fichier CSV :

df.to_csv('my_data.csv', columns=['col1', 'col4', 'col6'])

L’argument columns indique aux pandas quelles colonnes spécifiques exporter du DataFrame vers le fichier CSV.

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

Exemple : exporter des colonnes spécifiques de Pandas DataFrame vers un fichier CSV

Supposons que nous ayons le DataFrame pandas suivant qui contient des informations sur divers joueurs de basket-ball :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'team': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   'points': [18, 22, 19, 14, 14, 11, 20, 28],
                   'assists': [5, 7, 7, 9, 12, 9, 9, 4],
                   'rebounds': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
print(df)

  team  points  assists  rebounds
0    A      18        5        11
1    B      22        7         8
2    C      19        7        10
3    D      14        9         6
4    E      14       12         6
5    F      11        9         5
6    G      20        9         9
7    H      28        4        12

Si nous utilisons la fonction to_csv() pour exporter le DataFrame vers un fichier CSV, pandas exportera toutes les colonnes par défaut :

#export DataFrame to CSV file
df.to_csv('basketball_data.csv')

Voici à quoi ressemble le fichier CSV :

Notez que toutes les colonnes du DataFrame sont incluses dans le fichier CSV.

Pour exporter uniquement des colonnes spécifiques du DataFrame vers un fichier CSV, nous pouvons utiliser l’argument columns .

Par exemple, nous pouvons utiliser la syntaxe suivante pour exporter uniquement les colonnes équipe et rebonds vers le fichier CSV :

#export only team and rebounds columns from DataFrame to CSV file
df.to_csv('basketball_data.csv', columns=['team', 'rebounds'])

Voici à quoi ressemble le fichier CSV :

Notez que seules les colonnes équipe et rebonds sont incluses dans le fichier CSV.

Remarque : Vous pouvez trouver la documentation complète de la fonction pandas to_csv() ici .

Ressources additionnelles

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

Comment exporter des données vers un fichier CSV sans en-tête dans Pandas
Comment fusionner plusieurs fichiers CSV dans Pandas
Comment lire uniquement des lignes spécifiques d’un fichier CSV dans Pandas