可迭代对象、迭代器、生成器
可迭代对象
实现了__iter__方法的对象就叫做可迭代对象。
__iter__方法的作用就是返回一个迭代器对象。直观理解就是能用for循环进行迭代的对象就是可迭代对象。比如:字符串,列表,元祖,字典,集合等等,都是可迭代对象。
for循环与__iter()__方法的关系:
def generator():
yield 1
yield 2
yield 3
g = generator()
# print(next(g))
# print(next(g))
# print(next(g))
# print(next(g))
for i in g:
print(i)
调用可迭代对象的__iter__方法返回一个迭代器对象(iterator)
不断调用迭代器的__next__方法返回元素
直到迭代完成后,处理StopIteration异常
迭代器( iterator )
迭代器是一个带状态的对象,他能在你调用next()方法的时候返回下一个值,
__iter__返回迭代器自身,
__next__返回下一个值,如果没有更多元素了,则抛出StopIteration异常。
任何实现了__iter__和__next__()方法的对象都是迭代器,
根据定义,可以写一个迭代器,并通过next()方法来调用:
class MyIterator:
def __init__(self,begin,end):
self.begin = begin
self.end = end
def __iter__(self):
return self
def __next__(self):
if self.begin == self.end:
raise StopIteration('停止')
x = self.begin
self.begin += 1
return x
# iter_obj = MyIterator(1,4)
# #
# # print(next(iter_obj))
# # print(next(iter_obj))
# # print(next(iter_obj))
# # print(next(iter_obj))
for i in MyIterator(1,4):
print(i)
迭代器就像一个懒加载的工厂,等到有人需要的时候才给它生成值返回,没调用的时候就处于休眠状态等待下一次调用。
直到无元素可调用,返回StopIteration异常。
生成器(generator)
生成器其实是一种特殊的迭代器,不过这种迭代器更加优雅。它不需要再像上面的类一样写__iter__()和__next__()方法了,
只需要一个yiled关键字。 生成器一定是迭代器(反之不成立),因此任何生成器也是以一种懒加载的模式生成值。用生成器来实现斐波那契数列的例子是:
def my_range(begin,end):
i = begin
while i < end :
yield i
i += 1
for e in my_range(0,3):
print(e)
生成器特殊的地方在于函数体中没有return关键字,函数的返回值是一个生成器对象。当执行f=fib()返回的是一个生成器对象,此时函数体中的代码并不会执行,只有显示或隐示地调用next的时候才会真正执行里面的代码。
生成器还有一个send方法,可以往生成器里的变量传值:
send方法和next方法唯一的区别是在执行send方法会首先把上一次挂起的yield语句的返回值通过参数设定,从而实现与生成器方法的交互。
但是需要注意,在一个生成器对象没有执行next方法之前,由于没有yield语句被挂起,所以执行send方法会报错。
因为当send方法的参数为None时,它与next方法完全等价。但是注意,虽然这样的代码可以接受,但是不规范。所以,在调用send方法之前,还是先调用一次next方法为好。
def my_generator():
print('begin')
x = yield 1
print(x)
yield
g = my_generator()
a = next(g)
print(a)
b = g.send('sjz')
print(b)
生成器表达式
a = (e for e in range(10) )
print(a)
判断是否可以迭代
from collections.abc import Iterable
print(isinstance([], Iterable))
print(isinstance({}, Iterable))
print(isinstance('123', Iterable))
print(isinstance((a for a in range(10)), Iterable))
print(isinstance(123, Iterable))
#除了数字都可迭代