快捷導(dǎo)航

python中實(shí)現(xiàn)迭代器（iterator）的方法示例

更新時(shí)間：2017年01月19日 15:22:54 作者：瓜園耕讀

我們經(jīng)常需要遍歷一個(gè)對(duì)象中的元素，在Python中這種功能是通過(guò)迭代器來(lái)實(shí)現(xiàn)的。下面這篇文章主要給大家介紹了python中實(shí)現(xiàn)迭代器（iterator）的方法示例，需要的朋友可以參考借鑒，下面來(lái)一起看看吧。

概述

迭代器是訪問(wèn)集合元素的一種方式。迭代器對(duì)象從集合的第一個(gè)元素開(kāi)始訪問(wèn)，直到所有的元素被訪問(wèn)完結(jié)束。迭代器只能往前不會(huì)后退。

延遲計(jì)算或惰性求值 (Lazy evaluation)

迭代器不要求你事先準(zhǔn)備好整個(gè)迭代過(guò)程中所有的元素。僅僅是在迭代至某個(gè)元素時(shí)才計(jì)算該元素，而在這之前或之后，元素可以不存在或者被銷(xiāo)毀。這個(gè)特點(diǎn)使得它特別適合用于遍歷一些巨大的或是無(wú)限的集合。

今天創(chuàng)建了一個(gè)實(shí)體類(lèi)，大致如下：

class Account():
 def __init__(self,
   account_name,
   account_type,
   account_cost,
   return_amount=0):
 self.account_name = account_name # 賬戶名
 self.account_type = account_type # 賬戶類(lèi)型
 self.account_cost = account_cost # 月結(jié)費(fèi)用
 self.return_amount = return_amount # 返還金額

然后創(chuàng)建一個(gè)實(shí)體列表：

accounts = [Account("張三", "年費(fèi)用戶", 450.00, 50),
  Account("李四", "月結(jié)用戶", 100.00),
  Account("楊不悔", "月結(jié)用戶", 190.00, 25),
  Account("任我行", "月結(jié)用戶", 70.00, 10),
  Account("凌未風(fēng)", "年費(fèi)用戶", 400.00, 40)]

我想要執(zhí)行next()功能，也就是需要的時(shí)候“next”一下，得到List中的下一個(gè)元素。

直接測(cè)試一下：

結(jié)果發(fā)現(xiàn)List不支持next()特性。這時(shí)候，List只是一個(gè)iterable，而不是iterator。

iterable和iterator的區(qū)別如下：

iterable —— 只實(shí)現(xiàn)了__iter__的對(duì)象；
iterator —— 同時(shí)實(shí)現(xiàn)了__iter__和__next__方法的對(duì)象。

其中， __iter__返回iterator對(duì)象， __next__則返回迭代過(guò)程的下一個(gè)元素。

1. 讓列表成為iterator

要讓前面的accounts List成為iterator只需簡(jiǎn)單的一個(gè)iter()函數(shù)：

accounts_iterator = iter(accounts)
(next(accounts_iterator)).account_name

結(jié)果如下圖所示：

這么簡(jiǎn)單的函數(shù)，估計(jì)還是有不少Python開(kāi)發(fā)者不知道吧？

2. 自定義iterator對(duì)象

擴(kuò)展開(kāi)來(lái)講，如何定義自己的iterator對(duì)象呢？其實(shí)也就是按照上面的定義，實(shí)現(xiàn)__iter__和__next__方法。

我們接下來(lái)定義一個(gè)AccountIterator類(lèi)：

class AccountIterator():
 def __init__(self, accounts):
 self.accounts = accounts # 賬戶集合
 self.index = 0

 def __iter__(self):
 return self

 def __next__(self):
 if self.index >= len(self.accounts):
  raise StopIteration("到頭了...")
 else:
  self.index += 1
  return self.accounts[self.index-1]

運(yùn)行結(jié)果如：