如何理解python當中yield關鍵字的行爲

如何遍歷pandas當中的行

現有一個數據框pandas的dataframe:

import pandas as pd
inp = [{'c1':10, 'c2':100}, {'c1':11,'c2':110}, {'c1':12,'c2':120}]
df = pd.DataFrame(inp)
print df

指望輸出

c1   c2
0  10  100
1  11  110
2  12  120

如今我想遍歷這個數據框的每一行， 在每一行當中我想經過列名訪問第一行的每個元素，例如，實現如下效果：

for row in df.rows:
   print row['c1'], row['c2']

這裏推送一個很是好用的函數：
iterrows()
這個函數同時返回 索引和行對象的生成器能夠直接進行遍歷

In [18]: for index, row in df.iterrows():
   ....:     print row['c1'], row['c2']
   ....:     
10 100
11 110
12 120

如何理解python當中的yield函數

若要理解python當中的(yield)函數,首先必須理解什麼是生成器(generators),在理解生成器以前必須先理解迭代器(iterators).

迭代器

當你建立一個列表list,能夠一個元素一個元素逐個讀他,這樣的操做稱爲迭代 interation :

>>> mylist = [1, 2, 3]
>>> for i in mylist:
...    print(i)
1
2
3

這裏mylist就稱之爲一個可迭代對象,當你使用列表推導時(list comprehension),能夠生成一個列表,列表推導的方法以下所示:

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

全部能夠用for ... in ... 操做的對象稱之爲可迭代對象.例如字符串列表文件集合等等.

++列表這類可迭代對象還比較方便,但缺點就是須要存儲在內存中的對象很是多,在值很是多的時候若是都使用這種方式,不是很好++.

生成器

生成器也是迭代器的一種,是一種只能遍歷一次的可迭代對象.
生成器不須要在內存當中存儲全部的值,
他們是即時生成值,性能更快.
例如:

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

能夠看到,除了使用[]替代()以外,其它都是同樣的,可是如下的用法就是不容許的:

for i in mygenerator

實際上他們先生成0,而後忘掉0,再生成1,丟棄1,一直往下,一個接一個進行處理.

Yield函數

yield 函數有點像Return,區別在於這個函數返回的是迭代器.
例如:

def createGenerator():
    mylist = range(3)
    print u'this will be executed only when for ... in ..called only once'
    for i in mylist:
        yield i*i
mygenerator = createGenerator()
print 'test'
for i in mygenerator:
    print(i)



test
this will be executed only when for ... in ..called only once
0
1
4

能夠看出來,當你知道你的函數會返回數量很是大的元素供遍歷時,而且只需讀一次的時候,使用yield函數是很是合適的.

若要掌握yield函數,你必須理解當你調用這個函數時,函數內部的代碼實際是沒有執行的.這個函數只是返回一個生成器的對象.當實際遍歷時(for ... in ... ) yield語句纔會執行.

這裏是比較有意思的地方:
第一次使用for訪問這個生產器對象的時候,"print u'this will be executed only when for ... in ..called only once'"這句話才被打印出來,而且打印在"test"以後,說明yield以前的函數體會在for第一次循環時被調用一次且僅僅調用一次.
可是,若是是這樣狀況又不一樣了:

def createGenerator():
    mylist = range(3)
    print u'this will be executed only when for ... in ..called only once'
    for i in mylist:
        print 'test3'
        yield i*i
    print 'test2'
mygenerator = createGenerator()
print 'test'
for i in mygenerator:
    print(i)

結果如今變成了:

test
this will be executed only when for ... in ..called only once
test3
0
test3
1
test3
4
test2

可見若是在yield語句同級的代碼塊中的語句,其實外層for進行迭代時,每次都會執行.