Maison  >  Article  >  développement back-end  >  Comment exclure les en-têtes lors du traitement de fichiers CSV avec Python ?

Comment exclure les en-têtes lors du traitement de fichiers CSV avec Python ?

Barbara Streisand
Barbara Streisandoriginal
2024-10-31 18:28:29438parcourir

How to Exclude Headers When Processing CSV Files with Python?

Exclusion des en-têtes lors du traitement de fichiers CSV avec Python

Lors du traitement d'un fichier CSV, il peut être nécessaire d'exclure les en-têtes de certaines opérations. Ceci peut être réalisé en utilisant les modules de lecture et d'écriture CSV de Python.

Dans le code fourni, il est souhaité d'ignorer la ligne d'en-tête. Au lieu d'initialiser la variable de ligne à 1, une approche plus simple consiste à ignorer la première ligne avant de traiter le reste. Cela peut être accompli comme suit :

<code class="python">with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
    reader = csv.reader(infile)
    next(reader, None)  # Skip the headers
    writer = csv.writer(outfile)
    for row in reader:
        # Process each row
        writer.writerow(row)</code>

En appelant next(reader, None), la première ligne du fichier CSV est récupérée et supprimée. Les lignes suivantes peuvent ensuite être traitées et écrites dans le fichier de sortie.

De plus, le code peut être simplifié en utilisant des gestionnaires de contexte pour gérer automatiquement l'ouverture et la fermeture des fichiers :

<code class="python">with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
    reader = csv.reader(infile)
    writer = csv.writer(outfile)
    # Skip the headers
    if headers := next(reader, None):
        writer.writerow(headers)</code>

Dans ce code, la variable facultative d'en-têtes reçoit les en-têtes ignorés, permettant à l'utilisateur de les écrire sans traitement dans le fichier de sortie s'il le souhaite.

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