2016-09-16 148 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 –