Comment extraire des colonnes spécifiques d'un fichier CSV à l'aide de Python ?

Découper les colonnes CSV avec le module Python csv

Souvent, il est nécessaire d'analyser un fichier CSV (valeurs séparées par des virgules) et extraire des colonnes de données spécifiques. En utilisant le module csv de Python, on peut accomplir cette tâche facilement.

Extraction de colonnes

Pour extraire des colonnes spécifiques, utilisez la variable include_cols, qui spécifie les indices de colonne à compris. Chaque colonne est numérotée à partir de 0, la première colonne étant 0. Par exemple, pour capturer les colonnes « ID », « Nom », « Code postal » et « Téléphone », vous écririez Included_cols = [1, 2, 6 , 7].

Itération et impression

Pour parcourir les lignes et extraire les colonnes spécifiées, utilisez une compréhension de liste dans une boucle for. Pour chaque ligne, la variable de contenu sera renseignée avec les valeurs des colonnes spécifiées. Pour afficher les données extraites, utilisez print(content) dans la boucle.

included_cols = [1, 2, 6, 7]

for row in reader:
    content = list(row[i] for i in included_cols)
    print(content)

Optimisation avec Pandas

Pour une gestion CSV plus efficace, pensez à utiliser le module pandas . Pandas fournit des outils pratiques et puissants pour lire et manipuler les fichiers CSV. Pour lire un CSV et stocker une colonne spécifique dans une variable, utilisez simplement :

import pandas as pd
df = pd.read_csv(csv_file)
saved_column = df.column_name

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!