使用Python的CSV模块覆盖csv文件中的特定行

Question

我正在使用Python的csv模块来执行一些读取和写入csv文件。

我已经读得很好，追加到csv罚款，但我希望能够覆盖csv中的特定行。

供参考，这是我读，然后写代码来追加：

#reading 
    b = open("bottles.csv", "rb") 
    bottles = csv.reader(b) 
    bottle_list = [] 
    bottle_list.extend(bottles) 
    b.close() 

    #appending 
    b=open('bottles.csv','a') 
    writer = csv.writer(b) 
    writer.writerow([bottle,emptyButtonCount,100, img]) 
    b.close() 

而且我使用基本相同的覆盖模式（这是不正确的，它只是覆盖了整个csv文件）：

b=open('bottles.csv','wb') 
    writer = csv.writer(b) 
    writer.writerow([bottle,btlnum,100,img]) 
    b.close() 

在第二种情况下，我该如何告诉Python我需要一个特定的行被覆盖？我搜索了Gogle和其他*的帖子无济于事。我认为我有限的编程知识是责备而不是Google。

Answer 1

我将增加Steven答：

import csv 

bottle_list = [] 

# Read all data from the csv file. 
with open('a.csv', 'rb') as b: 
    bottles = csv.reader(b) 
    bottle_list.extend(bottles) 

# data to override in the format {line_num_to_override:data_to_write}. 
line_to_override = {1:['e', 'c', 'd'] } 

# Write data to the csv file and replace the lines in the line_to_override dict. 
with open('a.csv', 'wb') as b: 
    writer = csv.writer(b) 
    for line, row in enumerate(bottle_list): 
     data = line_to_override.get(line, row) 
     writer.writerow(data) 

Answer 2

您无法覆盖CSV文件中的单个行。您必须将所需的所有行写入新文件，然后将其重命名为原始文件名。

您的使用模式可能比CSV文件更适合数据库。查看sqlite3模块的轻量级数据库。