Comment extraire des données de colonnes spécifiques dans un fichier CSV avec le module CSV ?

Extraire des données de colonnes spécifiques dans un fichier CSV avec le module CSV

Lorsque vous travaillez avec des fichiers CSV, il est souvent nécessaire d'extraire des données à partir de colonnes spécifiques uniquement. Pour y parvenir à l'aide du module CSV, il est important de comprendre comment la bibliothèque gère la récupération des données.

Dans l'exemple de code fourni, la tentative d'accès à des colonnes spécifiques se fait via la syntaxe row[i], où i représente le numéro de colonne. Cependant, le résultat n'est pas celui attendu en raison d'une approche incorrecte.

La méthode correcte consiste à inclure l'instruction print dans la boucle for. Le code corrigé suivant le démontre :

import csv

included_cols = [1, 2, 6, 7]
with open("csv_file.csv") as csvfile:
    reader = csv.reader(csvfile, delimiter=",")
    for row in reader:
        content = list(row[i] for i in included_cols)
        print(content)  # Include the print statement within the loop to display each row

Maintenant, le code modifié affichera les valeurs des colonnes souhaitées pour chaque ligne.

Cependant, il convient de mentionner une approche alternative utilisant le bibliothèque pandas, qui simplifie la gestion des fichiers CSV. Avec pandas, vous pouvez facilement lire un fichier CSV et extraire des colonnes spécifiques dans des variables :

import pandas as pd

df = pd.read_csv("csv_file.csv")
names = df["Name"]  # Use df['column_name'] to extract a specific column into a variable

La bibliothèque pandas fournit un ensemble polyvalent d'outils pour travailler avec des données structurées, ce qui en fait un choix recommandé pour gérer les fichiers CSV. efficacement.

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!