如何使用python更新csv中行的特定列

Question

我有一個csv文件，其中包含以下內容

1,2,3,4 
a,b,c,d 
w,x,y,z 

我想將此csv文件內容更新爲

1,2,3,4 
a,b,c,d,k 
w,x,y,z 

有人可以分享這個更新過程的Python代碼嗎？

Answer 1

使用csv庫，我們讀取數據，將其轉換爲列表列表，將k附加到相關列表中，然後再次寫出。

import csv 

data = csv.reader(open("input.csv")) 

l = list(data)  
l[1].append("k") 

our_writer = csv.writer(open("output.csv", "wb")) 

our_writer.writerows(l) 

雖然CSV庫是不是你的玩具情況下完全必要的，它往往是很好的使用可很好的辦法。

Answer 2

你不需要csv這樣一個簡單的例子：

# sed '2s/$/,k/' 
import fileinput 
for lineno, line in enumerate(fileinput.input(inplace=True), start=1): 
    if lineno == 2: 
     line = line.rstrip('\n') + ",k\n" 
    print line, 

例子：

$ python update_2nd_line.py data_to_update.csv 

它更新data_to_update.csv文件就地。