可迭代物件

mylist 是一個可迭代的物件。當你使用一個列表生成式來建立一個列表的時候，就建立了一個可迭代的物件:

>>> mylist = [x*x for x in range(3)]
>>> for i in mylist :
...    print(i)
0
1
4

• 1
• 2
• 3
• 4
• 5
• 6

在這裡，所有的值都存在記憶體當中，所以並不適合大量資料

生成器

• 可迭代
• 只能讀取一次
• 實時生成資料，不全存在記憶體中

 >>> mygenerator = (x*x for x in range(3))
>>> for i in mygenerator :
...    print(i)
0
1
4
 

• 1
• 2
• 3
• 4
• 5
• 6

注意你之後不能再使用for i in mygenerator了

yield關鍵字

• yield 是一個類似 return 的關鍵字，只是這個函式返回的是個生成器
• 當你呼叫這個函式的時候，函式內部的程式碼並不立馬執行 ，這個函式只是返回一個生成器物件
• 當你使用for進行迭代的時候，函式中的程式碼才會執行

>>> def createGenerator() :
...    mylist = range(3)
...    for i in mylist :
...        yield i*i
...
>>> mygenerator = createGenerator() # create a generator
>>> print(mygenerator) # mygenerator is an object!
<generator object createGenerator at 0xb7555c34>
>>> for i in mygenerator:
...     print(i)
0
1
4
 

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• 11
• 12
• 13

第一次迭代中你的函式會執行，從開始到達 yield 關鍵字，然後返回 yield 後的值作為第一次迭代的返回值. 然後，每次執行這個函式都會繼續執行你在函式內部定義的那個迴圈的下一次，再返回那個值，直到沒有可以返回的。

控制生成器的窮盡

>>> class Bank(): # let's create a bank, building ATMs
...    crisis = False
...    def create_atm(self) :
...        while not self.crisis :
...            yield "$100"
>>> hsbc = Bank() # when everything's ok the ATM gives you as much as you want
>>> corner_street_atm = hsbc.create_atm()
>>> print(corner_street_atm.next())
$100
>>> print(corner_street_atm.next())
$100
>>> print([corner_street_atm.next() for cash in range(5)])
['$100', '$100', '$100', '$100', '$100']
>>> hsbc.crisis = True # crisis is coming, no more money!
>>> print(corner_street_atm.next())
<type 'exceptions.StopIteration'>
>>> wall_street_atm = hsbc.create_atm() # it's even true for new ATMs
>>> print(wall_street_atm.next())
<type 'exceptions.StopIteration'>
>>> hsbc.crisis = False # trouble is, even post-crisis the ATM remains empty
>>> print(corner_street_atm.next())
<type 'exceptions.StopIteration'>
>>> brand_new_atm = hsbc.create_atm() # build a new one to get back in business
>>> for cash in brand_new_atm :
...    print cash
$100
$100
$100
$100
$100
$100
$100
$100
$100
...