2016-09-16 146 views
0

下面是一個CSV文件的結構,我有,重命名CSV文件頭

ITEM | ID    | SET | COUNT | 
------ | ------   |------ | ------ | 
Value | 44000001005 | 0  | 24  | 
Value | 10000000019659 | 0  | 29  | 
Value | 10000000019659 | 1  | 5  | 

我要動態地取得來自用戶的輸入編輯的標題,所以,如果我認爲我已經採取「Edit_」作爲來自用戶的輸入,我的所得CSV文件看起來像下面,

EDIT_ITEM | EDIT_ID  | EDIT_SET | EDIT_COUNT | 
------------| ---------------|----------- | ------------| 
Value  | 44000001005 | 0   | 24   | 
Value  | 10000000019659 | 0   | 29   | 
Value  | 10000000019659 | 1   | 5   | 

因此,每個標頭值是由用戶輸入「Edit_」被附加。我可以使用「w.writerow」手動執行此操作,但不知道如何動態追加用戶輸入。

我對Python比較新,試圖在Stack Overflow上通過很多答案,但沒有得到任何可以用於我的東西。我希望對這個問題有任何指導。

謝謝!

回答

0

使用csv模塊來讀取和編輯頁眉那麼所有的行回寫到新的文件

import csv 
with open("input.csv") as f, open("output.csv", "wb") as o: 
    ui = raw_input("Enter user input: ") 
    reader = csv.reader(f)  
    header = reader.next() 
    writer = csv.writer(o) 
    new_header = map(lambda x:"{}{}".format(ui,x) ,header) 
    writer.writerow(new_header) 
    writer.writerows(reader) 
+0

謝謝你這麼多@Daniel。完美的作品! – RBK

+0

你介意接受答案@RaghavendraBhargava –