Python中的迭代器、生成器

from collections import Iterable, Iterator

1. 可迭代(iterable)对象

参考官网链接

class I:
def __init__(self, v):
self.v = v def __iter__(self):
return iter([self.v]) # 返回的应该是迭代器对象 i = I(2)
print(isinstance(i, Iterable), isinstance(i, Iterator)) # True False
print(iter(i)) # <list_iterator object at 0x00000000021C8358> class I:
def __init__(self, v):
self.v = v def __getitem__(self, item):
if item > self.v:
raise IndexError
return item i = I(2)
print(isinstance(i, Iterable), isinstance(i, Iterator)) # False False
print(iter(i)) # <iterator object at 0x0000000001DFCD30>

2. 迭代器(iterator)

参考官网链接

class I:
def __init__(self, v):
self.v = v def __iter__(self):
return self def __next__(self):
if self.v < 10:
self.v += 1
return self.v
raise StopIteration i = I(2)
print(isinstance(i, Iterable), isinstance(i, Iterator)) # True True
print(iter(i)) # <__main__.I object at 0x00000000021E0898>

3. 生成器(generator)

参考官网链接

class I:
def __init__(self, v):
self.v = v def __iter__(self):
yield 1 i = I(2)
print(isinstance(i, Iterable), isinstance(i, Iterator)) # True False
print(iter(i)) # <generator object I.__iter__ at 0x00000000022204C0>

4. 总结

  1. 只要实现了__iter__方法或者__getitem__方法, 就是iterable, 但Iterable只检测__iter__方法是否实现, 参见官网链接。推荐使用iter是否返回迭代器判断可否可迭代。
  2. __getitem__方法的返回有特殊要求, 参见官网链接
  3. 实现了__iter__和__next__方法就是一个iterator。参见pep234
  4. generator是返回一个generator iterator的函数。与普通函数的不同在于函数中包含 yield 表达式。

5. 其他参考

  1. Python可迭代与迭代器
  2. 迭代器
上一篇:Ng第十五课:异常检测(Anomaly Detection)


下一篇:DOM节点类型