自定義可迭代的類

列表可以獲取列表的長度，然后使用變量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 
          

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。