python可迭代对象、迭代器、生成器
可迭代对象iterable
一个拥有__iter__
方法的对象,可以使用for循环遍历
可迭代对象有: str
、list
、tuple
、dict
、set
、iterator
、generator
、file
# 判断一个对象是否可迭代
>>> from collections import Iterable
>>> isinstance('abc', Iterable)
True
>>> isinstance(100, Iterable)
False
迭代器iterator
- 一个实现了
__iter__
方法和__next__
方法的对象,就是迭代器 - 迭代器是可以被
next()
函数调用并不断返回下一个值的对象 iter()
函数通过调用可迭代对象的__iter__()
方法,获取该对象的迭代器,然后对获取到的迭代器不断使用next()
函数来获取下一条数据
迭代器有:generator
、iter(iterable)
>>> from collections import Iterator
>>> isinstance([1,2], Iterator) # 列表不是迭代器
False
>>> isinstance((1,2), Iterator)
False
>>> isinstance([i for i in range(10)], Iterator)
False
>>> isinstance((i for i in range(10)), Iterator)
True
>>> a = (i for i in range(3))
>>> next(a)
0
>>> next(a)
1
>>> next(a)
Traceback (most recent call last):
File "<input>", line 1, in <module>
NameError: name 'a' is not defined
>>> f = open('1.txt')
>>> isinstance(f, Iterator) # 文件是可迭代对象、也是迭代器
>>> True
# list、dict、str等虽然是可迭代对象,但不是迭代器,可以使用iter()函数转化
>>> isinstance('abc', Iterator)
False
>>> isinstance(iter('abc'), Iterator)
True
for循环遍历iterable的本质: 通过iter()函数获取iterable的迭代器,然后对获取到的迭代器不断调用next()函数获取下一个值,当遇到StopIteration的异常时循环结束
生成器generator
- 一边循环一边计算的机制。
- 一种特殊的迭代器,自动实现了"迭代器协议"(即
__iter__
和__next__
法), 生成器在迭代过程中可以改变当前迭代值 - 具有
yield
关键字的函数都是生成器,yield可以理解为return,返回后面的值给调用者。不同的是return返回后,函数会释放,而生成器则不会。在直接调用next方法或用for语句进行下一次迭代时,生成器会从yield下一句开始执行,直至遇到下一个yield。 - 生成器创建方法:
- 把一个列表生成式的
[]
中括号改为()
- 带有
yield
的函数
>>> li = [i for i in range(10)] # 列表生成式
>>> li
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> gen = (i for i in range(10)) # 迭代器
>>> gen
<generator object <genexpr> at 0x00000161109100F8>
>>> next(gen)
0
>>> next(gen)
1
实现斐波那契数列
def fib(max):
a, b, n = 0, 1, 0 # n表示数列的长度
while n < max:
# a, b =b, a+b
t = a + b
a = b
b = t
n += 1
yield a
num = fib(4)
print(next(num)) # 1
print(next(num)) # 1
print(next(num)) # 2
print(next(num)) # 3
print(next(num)) # StopIteration error