如何理解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进行迭代时,每次都会执行.