csv reader_Python中的for循环位置

Question

在我的这段脚本中，当计算 max_conc 时，下一个 for 循环不起作用，除非我把它放在阅读器后面。但是这样做， max_conc 不再起作用了。

with open ('outfile.csv', 'r') as infile:
    reader = csv.reader(infile)

    max_conc = max(cols[1] for cols in reader)
    print max_conc

    for row in reader:
    print row

重新打开输入文件是唯一的解决方案吗？

Answer 1

因为当你打电话时：

max(cols[1] for cols in reader)

你用尽了迭代器，你不能再循环遍历它。但是你可以使用seek 回到文件的开头：

infile.seek(0)

或者，您可以将文件的副本存储在内存中：

lines = infile.readlines()
max_conc = max(cols[1] for cols in lines)

另一种方式，正如@Lyncha 建议的那样，循环遍历文件的列表副本：

lines = list(reader)

这是一个小例子：

>>> f = open('test.txt')
>>> f.read()
'a\nb\nc\nd\ne\nf\ng'
>>> f.read()
''  #whoops, iterator is exhausted
>>> f.seek(0)
>>> f.read()
'a\nb\nc\nd\ne\nf\ng'