web-dev-qa-db-fra.com

Ajouter pythoniquement un en-tête à un fichier csv

J'ai écrit un script Python fusionnant deux fichiers CSV et je souhaite maintenant ajouter un en-tête au fichier CSV final. J'ai essayé de suivre les suggestions signalées ici et j'ai eu l'erreur suivante: expected string, float found. Quel est le moyen le plus pythonique de résoudre ce problème?

Voici le code que j'utilise:

import csv

with open('combined_file.csv', 'wb') as outcsv:
    writer = csv.DictWriter(outcsv, fieldnames = ["Date", "temperature 1", "Temperature 2"])
    writer.writeheader()

    with open('t1.csv', 'rb') as incsv:
        reader = csv.reader(incsv)
        writer.writerows(row + [0.0] for row in reader)

    with open('t2.csv', 'rb') as incsv:
        reader = csv.reader(incsv)
        writer.writerows(row[:1] + [0.0] + row[1:] for row in reader)
pythoncsvheader
48
albus_c

La classe DictWriter() attend dictionnaires pour chaque ligne. Si tout ce que vous vouliez faire était d'écrire un en-tête initial, utilisez une csv.writer() régulière et transmettez une ligne simple pour l'en-tête:

import csv

with open('combined_file.csv', 'wb') as outcsv:
    writer = csv.writer(outcsv)
    writer.writerow(["Date", "temperature 1", "Temperature 2"])

    with open('t1.csv', 'rb') as incsv:
        reader = csv.reader(incsv)
        writer.writerows(row + [0.0] for row in reader)

    with open('t2.csv', 'rb') as incsv:
        reader = csv.reader(incsv)
        writer.writerows(row[:1] + [0.0] + row[1:] for row in reader)

L'alternative serait de générer des dictionnaires lors de la copie de vos données:

import csv

with open('combined_file.csv', 'wb') as outcsv:
    writer = csv.DictWriter(outcsv, fieldnames = ["Date", "temperature 1", "Temperature 2"])
    writer.writeheader()

    with open('t1.csv', 'rb') as incsv:
        reader = csv.reader(incsv)
        writer.writerows({'Date': row[0], 'temperature 1': row[1], 'temperature 2': 0.0} for row in reader)

    with open('t2.csv', 'rb') as incsv:
        reader = csv.reader(incsv)
        writer.writerows({'Date': row[0], 'temperature 1': 0.0, 'temperature 2': row[1]} for row in reader)
73
Martijn Pieters

Vous ajoutez simplement une ligne supplémentaire avant d'exécuter la boucle. Cette ligne contient Votre nom d’en-tête de fichier CSV.

schema = ['a','b','c','b']
row = 4
generators = ['A','B','C','D']
with open('test.csv','wb') as csvfile:    
     writer = csv.writer(csvfile, delimiter=delimiter)
# Gives the header name row into csv
     writer.writerow([g for g in schema])   
#Data add in csv file       
     for x in xrange(rows):
         writer.writerow([g() for g in generators])
3
Mitul Panchal

Cela a fonctionné pour moi. 

header = ['row1', 'row2', 'row3']
some_list = [1, 2, 3]
with open('test.csv', 'wt', newline ='') as file:
    writer = csv.writer(file, delimiter=',')
    writer.writerow(i for i in header)
    for j in some_list:
        writer.writerow(j)
1
saggzz