Comment exclure les en-têtes lors du traitement de fichiers CSV avec 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!