如何使用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文件就地。