Python3之手動創(chuàng)建迭代器的實例代碼

更新時間：2019年05月22日 10:03:00 作者：若數

迭代器即可以遍歷諸如列表，字典及字符串等序列對象甚至自定義對象的對象，其本質就是記錄迭代對象中每個元素的位置。這篇文章主要介紹了Python3之手動創(chuàng)建迭代器,需要的朋友可以參考下

迭代器即可以遍歷諸如列表，字典及字符串等序列對象甚至自定義對象的對象，其本質就是記錄迭代對象中每個元素的位置。迭代過程從第一個元素至最后一個元素，且過程不能回滾或反方向迭代。

兩個基本方法iter、next

序列對象可以利用 iter() 直接創(chuàng)建迭代器，并通過 next() 即可迭代迭代器。

利用for循環(huán)迭代

S = 'PYTHON'
IT = iter(S)
for it in IT:
  print(it)

示例結果：

P
Y
T
H
O
N

利用next()迭代

S = 'PYTHON'
IT = iter(S)
print(next(IT))
print(next(IT))
print(next(IT))
print(next(IT))
print(next(IT))
print(next(IT))

示例結果：

P
Y
T
H
O
N

當我們使用next()迭代時，如果迭代了次數超過了迭代器中的元素個數就會引發(fā) StopIteration ，由此我們可以利用 while 循環(huán)迭代，并不斷捕捉迭代結束的異常完成for循環(huán)的迭代過程。

S = 'PYTHON'
IT = iter(S)
while True:
  try:
    print(next(IT))
  except StopIteration:
    break

P
Y
T
H
O
N

手動構建迭代器

把一個類作為一個迭代器使用需要在類中實現(xiàn)兩個方法 iter () 與 next () 。 iter () 方法返回一個特殊的迭代器對象，這個迭代器對象實現(xiàn)了 next () 方法并通過 StopIteration 異常標識迭代的完成。 next () 方法（Python 2 里是 next()）會返回下一個迭代器對象。下面這個類構造了一個迭代器接受一個可迭代的數字，每次迭代都返回上一次迭代結果的平方，當迭代結果大于 9999999999999 時拋出異常。利用該類創(chuàng)建示例對數字2進行平方和迭代。

class IT_SQUARE:
  def __init__(self, x):
    self.x = x

  def __next__(self):
    self.x = self.x ** 2
    if self.x > 9999999999999:
      raise StopIteration
    
    else:
      return self.x

  def __iter__(self):
    return self
 
IT1 = IT_SQUARE(2)
 
while True:
  try:
    print(IT1.__next__())
  except StopIteration:
    break

示例結果