小論python的迭代器和生成器

《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
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容