《python核心編程》的第一部分讀完之后,開始總結python的基礎結構圖,覺得生成器和迭代器的關系錯綜復雜,于是乎,沒事呀,就研究研究吧。
迭代器
內建函數iter()
list,tuple,dict,set,str等集合數據類型不是迭代器對象Iterator但它們是可迭代對象Iterable,可以使用iter()方法將Iterable變成Iterator對象
自定義迭代器
自定義迭代器必須實現__iter__()
和next()
方法
class d:
def __init__(self, data):
self.date = data
self.index = len(data)
def __iter__(self):
return self
def next(self):
if self.index == 0:
raise StopIteration
self.index = self.index - 1
return self.date[self.index]
dh = d([1, 5])
print type(dh) # <type 'instance'>
for x in dh:
print x
生成器
生成器是一種特殊的迭代器,內部支持了生成器協議,不需要明確定義iter()和next()方法
def b():
yield 12
yield 13
bf = b()
for i in bf:
print i
生成器中還有兩個很重要的方法:send()和close()。
send(value):
從前面了解到,next()方法可以恢復生成器狀態并繼續執行,其實send()是除next()外另一個恢復生成器的方法。
Python 2.5中,yield語句變成了yield表達式,也就是說yield可以有一個值,而這個值就是send()方法的參數,所以send(None)和next()是等效的。同樣,next()和send()的返回值都是yield語句處的參數(yielded value)
關于send()方法需要注意的是:調用send傳入非None值前,生成器必須處于掛起狀態,否則將拋出異常。也就是說,第一次調用時,要使用next()語句或send(None),因為沒有yield語句來接收這個值。
close():
這個方法用于關閉生成器,對關閉的生成器后再次調用next或send將拋出StopIteration異常。
生成器和迭代器主要有以下幾點區別:
- 迭代器是一個對象,生成器是一個函數。
- 迭代器包含
__iter__()
和next()
方法__iter__()
返回迭代器本身(self),next()返回next()方法返回容器的下一個元素;生成器有生成器函數生成,但是不用return返回,而是yield一次返回一個結果,且“暫停”代碼的執行。 - 生成器是一種特殊的迭代器,可使用next()方法。
def a():
yield 1
yield 2
af = a()
af.next() # 1
af.next() # 2