Maison  >  Article  >  développement back-end  >  Comment ajouter une nouvelle colonne à plusieurs fichiers CSV et aligner les entrées avec les lignes correspondantes à l'aide de Python ?

Comment ajouter une nouvelle colonne à plusieurs fichiers CSV et aligner les entrées avec les lignes correspondantes à l'aide de Python ?

DDD
DDDoriginal
2024-10-21 21:02:02593parcourir

How to Append a New Column to Multiple CSV Files and Align Entries with Corresponding Rows Using Python?

Ajout d'une nouvelle colonne aux fichiers CSV

Question initiale :

Les utilisateurs souhaitent ajouter une nouvelle colonne nommée "Berry" dans plusieurs fichiers CSV, en alignant les entrées avec les lignes correspondantes. Cependant, le script Python actuel saute des lignes et attribue uniquement « Berry » à la nouvelle colonne.

Solution :

  1. Obtenir et Modifier les en-têtes :

    • Récupérez la première ligne (en-têtes) à l'aide de next() (Python 3) :
    • Ajoutez l'en-tête "Berry" pour indiquer la nouvelle colonne.
  2. Itérer et modifier les lignes restantes :

    • Parcourez les lignes CSV restantes à l'aide d'un objet lecteur.
    • Pour chaque ligne, ajoutez la valeur de la première colonne à la nouvelle colonne "Berry".

Vous trouverez ci-dessous le script Python modifié intégrant ces modifications :

<code class="python">import csv

with open('input.csv', 'r') as csvinput, open('output.csv', 'w', newline='') as csvoutput:
    writer = csv.writer(csvoutput)
    reader = csv.reader(csvinput)

    headers = next(reader)
    headers.append('Berry')
    writer.writerow(headers)

    for row in reader:
        row.append(row[0])
        writer.writerow(row)</code>

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