Python: StopIteration异常和列表理解

Question

我想从csv文件中读取最多20行：

rows = [csvreader.next() for i in range(20)]

如果文件有20行或更多行，则工作正常，否则会失败，并引发StopIteration异常。

有没有一种优雅的方法来处理可能在列表理解中抛出StopIteration异常的迭代器，或者我应该使用常规的for循环？

Answer 1

您可以使用itertools.islice。它是列表切片的迭代器版本。如果迭代器的元素少于20个，它将返回所有元素。

import itertools
rows = list(itertools.islice(csvreader, 20))

Answer 2

itertools.izip (2)提供了一种轻松实现列表理解的方法，但在这种情况下，islice似乎是最佳选择。

from itertools import izip
[row for (row,i) in izip(csvreader, range(20))]

Answer 3

如果出于任何原因，您还需要跟踪行号，我建议您：

rows = zip(xrange(20), csvreader)

如果没有，你可以把它去掉，或者...那么，你最好从一开始就尝试其他更好的选择:-)