您好,登錄后才能下訂單哦!
自定義可迭代的類
列表可以獲取列表的長度,然后使用變量i對列表索引進行循環,也可以獲取集合的所有元素,且容易理解。沒錯,使用列表的代碼是容易理解,也很好操作,但這是要付出代價的。列表之所以可以用索引來快速定位其中的任何一個元素,是因為列表是一下子將所有的數據都裝載在內存中,而且是一塊連續的內存空間。當數據量比較小時,實現比較容易;當數據量非常大時,會非常消耗內存資源。而迭代就不同,迭代是讀取多少元素,就將多少元素裝載到內存中,不讀取就不裝載。這有點像處理XML的兩種方式:DOM和SAX。DOM是一下子將所有的XML數據都裝載到內存中,所以可以快速定位任何一個元素,但代價是消耗內存;而SAX是順序讀取XML文檔,沒讀到的XML文檔內容是不會裝載到內存中的,所以SAX比較節省內存,但只能從前向后的順序讀取XML文檔的內容。
如果在一個類中定義__iter__方法,那么這個類的實例就是一個迭代器。 __iter__方法需要返回一個迭代器,所以就返回對象本身即可(也就是self)。當對象每迭代一次時,就會調用迭代器中的另外一個特殊成員方法__next__ 。該方法需要返回當前迭代的結果。下面先看一個簡單的例子,在這個例子中,通過自定義迭代器對由星號*組成的直三角形的每一行進行迭代,然后通過for循環進行迭代,輸出一定行數的直角三角形。
# 可無限迭代直角三角形的行 class righttriangle: def __init__(self): # 定義一個變量n,表示當前的行數 self.n = 1 def __next__(self): # 通過字符串的乘法獲取直接三角形每一行的字符串,每一行字符串的長度是2 * n -1 result = '*' * (2 * self.n - 1) # 行數加1 self.n += 1 return result # 該方法必須返回一個迭代器 def __iter__(self): return self rt = righttriangle() # 對迭代器進行迭代 for e in rt: # 限制輸出行的長度不能大于20,否則將會無限輸出行 if len(e) > 20: break print(e)
輸出結果:
* *** ***** ******* ********* *********** ************* *************** ***************** *******************
[例10.10] 現在來看一個更有意思的例子,在這個例子中定義了一個迭代器類(Fibonacci),用于無限制迭代斐波那契數列。
# 可以無限制迭代斐波那契數列 class Fibonacci: # 在構造方法中定義兩個變量a和b,用來表示斐波那契數列的最開始的兩個值 def __init__(self): self.a = 0 self.b = 1 def __next__(self): # self.a就是當前要迭代的值 result = self.a # 計算斐波那契數列的下一個值,并將a變成原來的b,將b變成下一個值 self.a, self.b = self.b, self.a + self.b # 返回當前迭代的值 return result # 該方法必須返回一個迭代器 def __iter__(self): return self fibs = Fibonacci() # 對斐波那契數列進行迭代 for fib in fibs: print(fib,end = ' ') # 迭代的值不能超過500 if fib > 500: break
輸出結果:
0 1 1 2 3 5 8 13 21 34 55 89 144 233 377 610
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持億速云。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。