使用Python的CSV模块覆盖csv文件中的特定行
问题描述:
我正在使用Python的csv模块来执行一些读取和写入csv文件。使用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和其他stackoverflow的帖子无济于事。我认为我有限的编程知识是责备而不是Google。
答
我将增加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)
答
您无法覆盖CSV文件中的单个行。您必须将所需的所有行写入新文件,然后将其重命名为原始文件名。
您的使用模式可能比CSV文件更适合数据库。查看sqlite3模块的轻量级数据库。
+0
谢谢你的提示!我现在正在研究sqlite3文档。如果我有更多数据可用,我可能会使用它。现在奇点的解决方案很好,因为我的csv文件不是太大。 – 2010-11-11 13:39:09
谢谢!这完美的作品! – 2010-11-11 13:37:09