Я хотел бы прочитать не более 20 строк из файла CSV:Python: StopIteration исключение и списочные
rows = [csvreader.next() for i in range(20)]
отлично работает, если файл имеет 20 или более строк, терпит неудачу с исключением StopIteration иначе.
Есть ли элегантный способ справиться с итератором, который может вызвать исключение StopIteration в понимании списка или использовать регулярный цикл?
Вы можете использовать itertools.islice. Это итераторная версия списка. Если итератор имеет менее 20 элементов, он вернет все элементы.
import itertools
rows = list(itertools.islice(csvreader, 20))
Если по какой-либо причине вам необходимо также следить за номером строки, я бы порекомендовал вам:
rows = zip(xrange(20), csvreader)
Если нет, то вы можете лишить его после или ... ну, вы бы лучше попробовать другой вариант более оптимальный с самого начала :-)
Если вам нужен номер строки, вы должны использовать enumerate() .. –
itertools.izip (2) предоставляет возможность легко сделать списочные работать, но islice выглядит путь в этом случае.
from itertools import izip
[row for (row,i) in izip(csvreader, range(20))]
, который имеет то преимущество, что он не полагается на len() (например, для apsw.cursor) – Mark
'enumerate' - это правильный способ сделать это , без изменения диапазона. – ArekBulski
Thanks Ayman. Похоже, что переосмысления списков необходимо обновить, чтобы справиться с StopIteration, нет? Он кажется, что «для» уже обновлен, чтобы справиться с ним (он прекращает итерацию, когда сталкивается с этим исключением, неявно ловя его), и я не вижу очевидной причины, по которой переписные листы не будут делать то же самое. – Parand
для улавливания StopIteration относительно его итерации, а не других подобных объектов в его наборе. Так, например с = ITER (диапазон (5)) для г в диапазоне (10): \t печати I, c.next() поднимет StopIteration исключение относительно С. – Mapio
Цикл for НЕ НЕЯТНО улавливает StopIteration. Он только улавливает его, если его выбрасывает следующий метод итератора, а не если он забрасывается в тело цикла. В вашем вопросе csvreader.next() аналогичен телу цикла. – Miles