迭代器
迭代是訪問集合元素的一種方式索守。迭代器是一個可以記住遍歷的位置的對象访锻。迭代器對象從集合的第一個元素開始訪問话侄，直到所有的元素被訪問完結(jié)束。迭代器只能往前不會后退敛瓷。
- 可迭代對象
我們已經(jīng)知道可以對list叁巨、tuple、str等類型的數(shù)據(jù)使用for...in...的循環(huán)語法從其中依次拿到數(shù)據(jù)進行使用呐籽，我們把這樣的過程稱為遍歷锋勺，也叫迭代。
但是绝淡，是否所有的數(shù)據(jù)類型都可以放到for...in...的語句中宙刘，然后讓for...in...每次從中取出一條數(shù)據(jù)供我們使用，即供我們迭代嗎牢酵？
for i in 100:
... print(i)
...
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: 'int' object is not iterable
int整型不是iterable悬包，即int整型不是可以迭代的
我們自定義一個容器MyList用來存放數(shù)據(jù)，可以通過add方法向其中添加數(shù)據(jù)
class MyList(object):
... def init(self):
... self.container = []
... def add(self, item):
... self.container.append(item)
...
mylist = MyList()
mylist.add(1)
mylist.add(2)
mylist.add(3)
for num in mylist:
... print(num)
...
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: 'MyList' object is not iterable
MyList容器的對象也是不能迭代的
我們自定義了一個容器類型MyList馍乙，在將一個存放了多個數(shù)據(jù)的MyList對象放到for...in...的語句中布近，發(fā)現(xiàn)for...in...并不能從中依次取出一條數(shù)據(jù)返回給我們，也就說我們隨便封裝了一個可以存放多條數(shù)據(jù)的類型卻并不能被迭代使用丝格。
我們把可以通過for...in...這類語句迭代讀取一條數(shù)據(jù)供我們使用的對象稱之為可迭代對象（Iterable）**撑瞧。
- 如何判斷一個對象是否可以迭代
可以使用 isinstance() 判斷一個對象是否是 Iterable 對象：
In [50]: from collections import Iterable
In [51]: isinstance([], Iterable)
Out[51]: True
In [52]: isinstance({}, Iterable)
Out[52]: True
In [53]: isinstance('abc', Iterable)
Out[53]: True
In [54]: isinstance(mylist, Iterable)
Out[54]: False
In [55]: isinstance(100, Iterable)
Out[55]: False
- 可迭代對象的本質(zhì)
我們分析對可迭代對象進行迭代使用的過程，發(fā)現(xiàn)每迭代一次（即在for...in...中每循環(huán)一次）都會返回對象中的下一條數(shù)據(jù)显蝌，一直向后讀取數(shù)據(jù)直到迭代了所有數(shù)據(jù)后結(jié)束预伺。那么订咸，在這個過程中就應(yīng)該有一個“人”去記錄每次訪問到了第幾條數(shù)據(jù)，以便每次迭代都可以返回下一條數(shù)據(jù)酬诀。我們把這個能幫助我們進行數(shù)據(jù)迭代的“人”稱為迭代器(Iterator)脏嚷。
可迭代對象的本質(zhì)就是可以向我們提供一個這樣的中間“人”即迭代器幫助我們對其進行迭代遍歷使用。
可迭代對象通過iter方法向我們提供一個迭代器瞒御，我們在迭代一個可迭代對象的時候父叙，實際上就是先獲取該對象提供的一個迭代器，然后通過這個迭代器來依次獲取對象中的每一個數(shù)據(jù).
那么也就是說肴裙，一個具備了iter方法的對象趾唱，就是一個可迭代對象。
class MyList(object):
... def init(self):
... self.container = []
... def add(self, item):
... self.container.append(item)
... def iter(self):
... """返回一個迭代器"""
... # 我們暫時忽略如何構(gòu)造一個迭代器對象
... pass
...
mylist = MyList()
from collections import Iterable
isinstance(mylist, Iterable)
True
這回測試發(fā)現(xiàn)添加了iter方法的mylist對象已經(jīng)是一個可迭代對象了
- iter()函數(shù)與next()函數(shù)
list蜻懦、tuple等都是可迭代對象甜癞，我們可以通過iter()函數(shù)獲取這些可迭代對象的迭代器。然后我們可以對獲取到的迭代器不斷使用next()函數(shù)來獲取下一條數(shù)據(jù)阻肩。iter()函數(shù)實際上就是調(diào)用了可迭代對象的iter方法带欢。
li = [11, 22, 33, 44, 55]
li_iter = iter(li)
next(li_iter)
11
next(li_iter)
22
next(li_iter)
33
next(li_iter)
44
next(li_iter)
55
next(li_iter)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
StopIteration
注意，當我們已經(jīng)迭代完最后一個數(shù)據(jù)之后烤惊，再次調(diào)用next()函數(shù)會拋出StopIteration的異常乔煞，來告訴我們所有數(shù)據(jù)都已迭代完成，不用再執(zhí)行next()函數(shù)了柒室。
- 如何判斷一個對象是否是迭代器
可以使用 isinstance() 判斷一個對象是否是 Iterator 對象：
In [56]: from collections import Iterator
In [57]: isinstance([], Iterator)
Out[57]: False
In [58]: isinstance(iter([]), Iterator)
Out[58]: True
In [59]: isinstance(iter("abc"), Iterator)
Out[59]: True
- 迭代器Iterator
通過上面的分析渡贾，我們已經(jīng)知道，迭代器是用來幫助我們記錄每次迭代訪問到的位置雄右，當我們對迭代器使用next()函數(shù)的時候空骚，迭代器會向我們返回它所記錄位置的下一個位置的數(shù)據(jù)。實際上擂仍，在使用next()函數(shù)的時候囤屹，調(diào)用的就是迭代器對象的next方法（Python3中是對象的next方法，Python2中是對象的next()方法）逢渔。所以肋坚，我們要想構(gòu)造一個迭代器，就要實現(xiàn)它的next方法肃廓。但這還不夠智厌，python要求迭代器本身也是可迭代的，所以我們還要為迭代器實現(xiàn)iter方法盲赊，而iter方法要返回一個迭代器铣鹏，迭代器自身正是一個迭代器，所以迭代器的iter方法返回自身即可哀蘑。
一個實現(xiàn)了iter方法和next方法的對象诚卸，就是迭代器葵第。
class MyList(object):
"""自定義的一個可迭代對象"""
def init(self):
self.items = []
def add(self, val):
self.items.append(val)
def __iter__(self):
myiterator = MyIterator(self)
return myiterator
class MyIterator(object):
"""自定義的供上面可迭代對象使用的一個迭代器"""
def init(self, mylist):
self.mylist = mylist
# current用來記錄當前訪問到的位置
self.current = 0
def __next__(self):
if self.current < len(self.mylist.items):
item = self.mylist.items[self.current]
self.current += 1
return item
else:
raise StopIteration
def __iter__(self):
return self
if name == 'main':
mylist = MyList()
mylist.add(1)
mylist.add(2)
mylist.add(3)
mylist.add(4)
mylist.add(5)
for num in mylist:
print(num)
- for...in...循環(huán)的本質(zhì)
for item in Iterable 循環(huán)的本質(zhì)就是先通過iter()函數(shù)獲取可迭代對象Iterable的迭代器，然后對獲取到的迭代器不斷調(diào)用next()方法來獲取下一個值并將其賦值給item合溺，當遇到StopIteration的異常后循環(huán)結(jié)束羹幸。
1、先判斷Iterble是否可迭代
2辫愉、在第一步成立的情況下調(diào)用iter函數(shù) 得到Iterble對象的iter方法的返回值
3、iter方法的返回值是一個迭代器
可迭代對象里面必須要iter方法将硝，返回的必需是一個迭代器恭朗，迭代器里面必需有兩個方法 iter next
for去取的是迭代器里面的next的返回值。
- 迭代器的應(yīng)用場景
我們發(fā)現(xiàn)迭代器最核心的功能就是可以通過next()函數(shù)的調(diào)用來返回下一個數(shù)據(jù)值依疼。如果每次返回的數(shù)據(jù)值不是在一個已有的數(shù)據(jù)集合中讀取的痰腮，而是通過程序按照一定的規(guī)律計算生成的，那么也就意味著可以不用再依賴一個已有的數(shù)據(jù)集合律罢，也就是說不用再將所有要迭代的數(shù)據(jù)都一次性緩存下來供后續(xù)依次讀取膀值，這樣可以節(jié)省大量的存儲（內(nèi)存）空間。
舉個例子误辑，比如沧踏，數(shù)學(xué)中有個著名的斐波拉契數(shù)列（Fibonacci），數(shù)列中第一個數(shù)為0巾钉，第二個數(shù)為1翘狱，其后的每一個數(shù)都可由前兩個數(shù)相加得到：
0, 1, 1, 2, 3, 5, 8, 13, 21, 34, ...
現(xiàn)在我們想要通過for...in...循環(huán)來遍歷迭代斐波那契數(shù)列中的前n個數(shù)。那么這個斐波那契數(shù)列我們就可以用迭代器來實現(xiàn)砰苍，每次迭代都通過數(shù)學(xué)計算來生成下一個數(shù)潦匈。
class FibIterator(object):
"""斐波那契數(shù)列迭代器"""
def init(self, n):
"""
:param n: int, 指明生成數(shù)列的前n個數(shù)
"""
self.n = n
# current用來保存當前生成到數(shù)列中的第幾個數(shù)了
self.current = 0
# num1用來保存前前一個數(shù)，初始值為數(shù)列中的第一個數(shù)0
self.num1 = 0
# num2用來保存前一個數(shù)赚导，初始值為數(shù)列中的第二個數(shù)1
self.num2 = 1
def __next__(self):
"""被next()函數(shù)調(diào)用來獲取下一個數(shù)"""
if self.current < self.n:
num = self.num1
self.num1, self.num2 = self.num2, self.num1+self.num2
self.current += 1
return num
else:
raise StopIteration
def __iter__(self):
"""迭代器的__iter__返回自身即可"""
return self
if name == 'main':
fib = FibIterator(10)
for num in fib:
print(num, end=" ")
- 并不是只有for循環(huán)能接收可迭代對象
除了for循環(huán)能接收可迭代對象茬缩，list、tuple等也能接收吼旧。
li = list(FibIterator(15))
print(li)
tp = tuple(FibIterator(6))
print(tp)
