Python Chỉnh sửa tiêu đề CSV

Question

Tôi có dữ liệu sau đây từ tệp csv có tên là 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 

tôi cần phải thay đổi các tiêu đề để đọc

Item Number,Item Description,List Price,QTY Available 

Tôi đã được tìm kiếm câu hỏi tương tự trên đây và có không phải là một giải pháp mà tôi có thể hiểu được vì tôi tương đối mới để lập trình python. Cho đến nay tôi có:

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) 

Tôi chỉ đọc tệp gốc và sau đó ghi vào _modified. Làm cách nào để chọn các tiêu đề hiện tại và sau đó thay đổi chúng để chúng ghi vào tệp mới?

Answer 1

Tiêu đề chỉ là một hàng dữ liệu CSV khác. Chỉ cần viết chúng dưới dạng hàng mới vào đầu ra, tiếp theo là phần còn lại của dữ liệu từ tệp đầu vào.

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) 

Đối với Python 3, sử dụng:

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

và bạn có thể phải chỉ định một mã hóa cho dữ liệu của bạn.

Answer 2

Bạn có thể sử dụng fileinput cho việc này:

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

Một giải pháp khác là sử dụng các mô-đun fileinput để cập nhật các tập tin tại chỗ:

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,