python iter函数特殊用法

来源:互联网 发布:阿拉丁控股集团 知乎 编辑:程序博客网 时间:2024/06/05 01:05

Python 中迭代对象时会调用 iter(x)。
iter 函数还有一个鲜为人知的用法: 传入两个参数, 使用常规的函数或任何可调用的对象创建迭代器。
第一个参数必须是可调用的对象,用于不断调用(没有参数),产出各个值。
第二个值是哨符,这是个标记值, 当可调用的对象返回这个值时, 触发迭代器抛出 StopIteration 异常,而不产出哨符。

from random import randint>>> def random_num():...     return randint(1, 6)... >>> num = iter(random_num, 1)>>> num<callable-iterator object at 0x263aad0>>>> for i in num:...     print(i)... 446664
这里iter函数返回一个callable_iterator对象。1是哨符,对象返回这个值时触发迭代器抛出 StopIteration 异常。循环结束 与常规的迭代器一样,示例中的num对象一旦耗尽就没用了。