在Python中,一切皆对象
- Iterable 可迭代对象
- Iterator 迭代器
Iterable 凡是可以for循环的,都属于可迭代对象
Iterator 凡是可以next()或者__next__()的,都是迭代器
可迭代对象不一定是迭代器,迭代器一定是可迭代对象
只拥有__iter__()的对象是可迭代对象
拥有__iter__()和__next__()的对象是迭代器
可迭代协议就是内部实现__iter__()
迭代器协议必须拥有__iter__()和__next__()
- iter()和__iter__()的作用是返回一个迭代器
- 迭代器用next()和__next__()返回迭代器中的下一个对象,当没有时会返回StopIteration错误
from collections import Iterator,Iterable
a = [1,2,3,4]
print(isinstance(a,Iterable)) # True
print(isinstance(a,Iterator)) # False
b = iter(a)
print(isinstance(b,Iterable)) # True
print(isinstance(b,Iterator)) # True
print(b.__next__()) # 1
print(next(b)) # 2
c = a.__iter__()
print(next(c)) # 1
print(c.__next__()) # 2
迭代器的作用
迭代器可以节约内存
用迭代器实现斐波那契数列
class Fib():
def __init__(self,num):
self.num = num
self.a = 0
self.b = 1
self.current = 1
def __iter__(self):
return self
def __next__(self):
if self.current <= self.num:
self.a , self.b = self.b , self.a + self.b
self.current += 1
return self.a
else:
raise StopIteration
for i in Fib(5):
print(i,end=" ") # 1 1 2 3 5
for循环原理
- for语句内部会先实现__iter__(),将对象变成一个可迭代对象
- 然后实现__next__()方法,返回一个迭代器
- 然后调用__next__()获取值,当无值可取后会报一个StopIteration错误,for循环捕捉到错误停止循环
(从上面斐波那契数列的实现可以看出)