问题描述
我找到了 this example of a simple CSV read / write。
import csv
with open('input.csv',newline='') as csvfile:
rdr = csv.reader(csvfile)
l = sorted(rdr,key=lambda x: x[6],reverse=True)
with open('output.csv','w') as csvout:
wrtr = csv.writer(csvout)
wrtr.writerows(l)
上面的代码将一个 CSV 文件输出到 output.csv 中
如果我需要进行排序以外的其他处理,我可能会通过遍历 CSV 对象的行来完成。展示问题的最简单示例就是迭代行并打印它们:
import csv
with open('input.csv',newline='') as csvfile:
rdr = csv.reader(csvfile)
for arow in rdr:
print(arow)
l = sorted(rdr,'w') as csvout:
wrtr = csv.writer(csvout)
wrtr.writerows(l)
上面的代码按预期将 CSV 行打印到屏幕上。 但是,output.csv 文件然后是空的(0 字节)
为什么遍历行会导致 csv.writer 没有输出? CSV 对象中是否存在某种类型的持久状态需要重置?
解决方法
正如@Barmar 正确指出的:
rdr 是一个迭代器。一个迭代器只能循环一次。 如果要多次处理,请将其转换为列表。
工作代码:
import csv
with open('input.csv',newline='') as csvfile:
rdr = csv.reader(csvfile)
mylist = list(rdr)
for arow in mylist:
print(arow)
l = sorted(mylist,key=lambda x: x[6],reverse=True)
with open('output.csv','w') as csvout:
wrtr = csv.writer(csvout)
wrtr.writerows(l)