csv.reader / writer - 为什么迭代 CSV 对象会导致 writer 没有输出？

如何解决csv.reader / writer - 为什么迭代 CSV 对象会导致 writer 没有输出？

我找到了 this example of a simple CSV read / write。

此代码读取和写入 CSV 文件，按第 7 列排序。

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)