iter (noun, genitive itineris); n, third declension
Обект с метод next()
, който:
StopIteration
Вградената функция, която приема един аргумент и връща итератор за него. Работи както очаквате за вградети типове - list
, tuple
.
>>> i = iter([1, 2])
>>> i.next()
1
>>> i.next()
2
>>> i.next()
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
StopIteration
for target in sequence: блок
iter(sequence)
за да вземе итераторnext()
върху него. Прекратява ако той предизвика StopIteration
.target
към върната стойност и изпълнява блок
Ако искаме наш обект да може да бъде итериран чрез for
имаме избор:
__getitem__
__iter__
Ако изберете този подход, __iter__
трябва да връща обект, който има метод next()
. Вече знаете как той трябва да работи
class FileIter(object): def __init__(self, file): self.file = file def next(self): while(True): next_line = self.file.readline() if not next_line: raise StopIteration if not next_line.startswith("#"): return next_line def __iter__(self): return self print FileIter
<class '__main__.FileIter'>
Ако не сте дефинирали __iter__
, тогава функцията iter()
ще провери дали вашия обект няма __getitems__
. Ако да, тя сама конструира итератор, работещ така:
IndexError
.class Vector(object): def __init__(self, x, y, z): self.x = x self.y = y self.z = z def __getitem__(self, index): if index == 0: return self.x elif index == 1: return self.y elif index == 2: return self.z else: raise IndexError
Структурата на итераторите ни позволява…
Но:
Има, разбира се!
def no_comment(file): while(True): line = file.readline() if not line: return if not line.startswith("#"): yield line
yield
се наричат генератори. Те са по-особени.next()
.next()
, генератора се изпълнява като нормална функция.yield
, изпълнението на функцията замръзва и next()
връща стойността предадена на yield
.next()
върху обекта-генератор функцията продължава от същото място по същия начин.next()
вдига StopIteration
def flat(collection): for element in collection: if isinstance(element, (list, tuple)): for x in flat(element): yield x else: yield element print list(flat([1, 2, [3, 4, 5], 5, [6, [7, [8, 9], 10], 11]]))
[1, 2, 3, 4, 5, 5, 6, 7, 8, 9, 10, 11]
def fib(): x, y = 0, 1 while(True): yield y x, y = y, x + y