Python Edytuj nagłówki CSV

Question

Mam następujące dane z pliku csv o nazwie temp.

Item,Description,Base Price,Available 
2000-000-000-300,AC - CF/M Series Green For Black Hood,299.99,3 
2000-000-000-380,AC - CF/M Series Green For White Hood,299.99,3 

muszę zmienić nagłówki czytać

Item Number,Item Description,List Price,QTY Available 

I zostały poszukiwania podobnych pytań na ten temat tutaj i nie mają rozwiązania, które mogę zrozumieć, ponieważ jestem stosunkowo nowy do programowania Python. Do tej pory mam:

import csv 
import os 

inputFileName = "temp.csv" 
outputFileName = os.path.splitext(inputFileName)[0] + "_modified.csv" 

with open(inputFileName) as inFile, open(outputFileName, "w") as outfile: 
    r = csv.reader(inFile) 
    w = csv.writer(outfile) 

Co wiem, tylko odczytuje oryginalny plik, a następnie napiszę do _modyfikowanego. Jak wybrać bieżące nagłówki, a następnie zmienić je tak, aby zapisywały się do nowego pliku?

Answer 1

Nagłówki to tylko kolejny wiersz danych CSV. Po prostu zapisz je jako nowy wiersz do danych wyjściowych, a następnie pozostałe dane z pliku wejściowego.

import csv 
import os 

inputFileName = "temp.csv" 
outputFileName = os.path.splitext(inputFileName)[0] + "_modified.csv" 

with open(inputFileName, 'rb') as inFile, open(outputFileName, 'wb') as outfile: 
    r = csv.reader(inFile) 
    w = csv.writer(outfile) 

    next(r, None) # skip the first row from the reader, the old header 
    # write new header 
    w.writerow(['Item Number', 'Item Description', 'List Price', 'QTY Available']) 

    # copy the rest 
    for row in r: 
     w.writerow(row) 

dla Pythona 3, należy:

with open(inputFileName, newline='') as inFile, open(outputFileName, 'w', newline='') as outfile: 

i być może trzeba będzie określić kodowanie danych.

Answer 2

Można użyć fileinput na to:

import fileinput 
import sys 
import os 

inputFileName = "temp.csv" 
outputFileName = os.path.splitext(inputFileName)[0] + "_modified.csv" 

with open(outputFileName, "w") as outfile: 
    for line in fileinput.input(
     [inputFileName], 
     inplace=False): 
     if fileinput.isfirstline(): 
      outfile.write('Item Number,Item Description,List Price,QTY Available\n') 
     else: 
      outfile.write(line) 

Answer 3

Innym rozwiązaniem jest użycie modułu fileinput zaktualizować plik w lokalu:

import fileinput 
for line in fileinput.input('temp', inplace=True): 
    if fileinput.isfirstline(): 
     print 'Item Number,Item Description,List Price,QTY Available' 
    else: 
     print line,