python中的yield_如何理解python当中yield关键字的行为

最新推荐文章于 2022-11-22 01:06:23 发布

weixin_39526185

最新推荐文章于 2022-11-22 01:06:23 发布

阅读量103

点赞数

文章标签： 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)

这里mylist就称之为一个可迭代对象,当你使用列表推导时(list comprehension),可以生成一个列表,列表推导的方法如下所示:

>>> mylist = [x*x for x in range(3)]

>>> for i in mylist:

... print(i)

所有可以用for ... in ... 操作的对象称之为可迭代对象.例如字符串列表文件集合等等.

++列表这类可迭代对象还比较方便,但缺点就是需要存储在内存中的对象非常多,在值非常多的时候如果都使用这种方式,不是很好++.

生成器

生成器也是迭代器的一种,是一种只能遍历一次的可迭代对象.

生成器不需要在内存当中存储所有的值,

他们是即时生成值,性能更快.

例如:

>>> mygenerator = (x*x for x in range(3))

>>> for i in mygenerator:

... print(i)

可以看到,除了使用[]替代()之外,其它都是一样的,但是以下的用法就是不允许的:

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

可以看出来,当你知道你的函数会返回数量非常大的元素供遍历时,并且只需读一次的时候,使用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)