python - 用python编辑csv中的行

Question

到目前为止，这是我的代码：

def edit_assignment():
    print 'Select Assignment'
    list_assignment()
    f1 = open(fdir, 'r')
    lines = f1.readlines()
    f1.close()
    line_to_edit = raw_input('Assignment to edit(start from 0): ')
    if os.path.exists(fdir):
        f = open(fdir, "w")
        t1 = raw_input('TEST:')
        lines[int(line_to_edit)] 
        f.writelines()
        f.close()        
    else:
        check()
        edit_assignment()

这是较大代码的一部分，但我不知道如何获取它来替换 csv 中的第 X 行。我已经坚持了一段时间，不知道该去哪里。

任何帮助表示赞赏

Answer 1

可以改进此代码，但它确实有效。只需更改output_array[row_to_edit]您要编辑的行和相应的值，然后它将使用该数据重写 CSV。

import csv

c = csv.reader(open('YourCSV.csv', 'rU'))

output_array = []
for row in c:
    output_array.append(row)

output_array[row_to_edit] = 'Edited'

writer = csv.writer(open('YourCSV.csv', 'wb'))

for row in output_array:
    writer.writerow(row)