Maison >développement back-end >Tutoriel Python >Comment extraire des données de colonnes spécifiques dans un fichier CSV à l'aide du module « csv » de Python ?

Comment extraire des données de colonnes spécifiques dans un fichier CSV à l'aide du module « csv » de Python ?

Mary-Kate Olsen
Mary-Kate Olsenoriginal
2024-11-15 22:48:02652parcourir

How to Extract Data from Specific Columns in a CSV File Using Python's 'csv' Module?

Comment extraire des données de colonnes spécifiques d'un fichier CSV à l'aide du module 'csv'

Dans cet article, nous aborderons la question de extraire des données de colonnes spécifiques d'un fichier CSV à l'aide du Python 'csv' module.

Problème :

Vous essayez d'analyser un fichier CSV et de récupérer les données de colonnes particulières uniquement (par exemple, ID, Nom, Code postal et Téléphone). Cependant, votre code ne produit que la dernière colonne.

Solution :

La principale erreur dans le code fourni était l'emplacement de l'instruction « print ». Il doit être à l'intérieur de la « boucle for » :

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

Solution alternative utilisant Pandas :

Le module Pandas fournit une solution plus efficace et plus élégante pour gérer les fichiers CSV :

import pandas as pd

df = pd.read_csv(csv_file)
names = df['Name']
print(names)  # prints all the names

Dans ce cas, la colonne 'Nom' est enregistrée dans les 'noms' variable.

Autres considérations :

  • Assurez-vous que la liste « inclus_cols » contient les numéros de colonne corrects (à partir de 0).
  • Vérifiez les erreurs d'indentation dans le code, car elles peuvent provoquer un comportement imprévisible.
  • Si l'instruction 'print' est à l'intérieur de la boucle mais n'affiche toujours que la dernière colonne, cela pourrait être dû à un problème avec le fichier CSV lui-même.

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!

Déclaration:
Le contenu de cet article est volontairement contribué par les internautes et les droits d'auteur appartiennent à l'auteur original. Ce site n'assume aucune responsabilité légale correspondante. Si vous trouvez un contenu suspecté de plagiat ou de contrefaçon, veuillez contacter admin@php.cn