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