Python yield使用方法示例

2018-09-23 20:26

阅读:525

  1. iterator
叠代器最简单例子应该是数组下标了,且看下面的c++代码:

  复制代码 代码如下:
int array[10];
for ( int i = 0; i < 10; i++ )
printf(%d , array[i]);

  叠代器工作在一个容器里(array[10]),它按一定顺序(i++)从容器里取出值(array[i])并进行操作(printf(%d , array[i])。

  上面的代码翻译成python:

  复制代码 代码如下:
array = [i for i in range(10)]
for i in array:
print i,

  首先,array作为一个list是个容器,其次list这个内建类型有默认的next行为,python发现这些之后采取的秘密的没被各位看到的动作是:拿出array这丫容器的叠代器,从里面next一下把值给i供for循环主体处置,for把这个值print了。

现在的问题是数据可以做容器叠代,代码可以吗?

2. constructor

  怎么把函数变成constructor? 在函数体里有yield就行了!

  复制代码 代码如下:
def gen():
print enter
yield 1
print next
yield 2
print next again

  for i in gen():
print i

  各位!python看到gen函数里出现yield,知道可以用next了,问题是怎么对代码这个容器玩next?
从容器里拿到iterator的时候它还什么也不是,处在容器入口处,对于数组来说就是下标为-1的地方,对于函数来说就是函数入口嘛事没干,但是万事俱备就欠next。
开始for i in g,next让itreator爬行到yield语句存在的地方并返回值,
再次next就再爬到下一个yield语句存在的地方并返回值,依次这样直到函数返回(容器尽头)。
您一定看出来上面代码的输出是:
enter
1
next
2
next again

3. 使用yield
yield的代码叠代能力不但能打断函数执行还能记下断点处的数据,下次next书接上回,
这正是递归函数需要的。
例如中序遍历二叉树:
(应该是David Mertz写的)

  复制代码 代码如下:
def inorder(t):
if t:
for x in inorder(t.left):
yield x
for x in inorder(t.right):
yield x
for n in inorder(tree)
print n


评论


亲,登录后才可以留言!