欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python中的itertools与iter

程序员文章站 2024-02-12 13:27:16
...

ptyhon中的itertools模块的使用

无限迭代器

这三个方法会生成一个无限的迭代器,也就是说,如果你用for进行访问的化,程序不会终止,只有在按下ctrl+c的时候才会结束,如果一直不结束就会导致系统内存被占满而引起系统崩溃

名称 Arguments result annotation
count() start, [step] start, start+step, start+2*step, … 与range()类似
cycle() p p0, p1, … plast, p0, p1, … 循环输出p中的元素
repeat() elem [,n] elem, elem, elem, … endlessly or up to n times 输出n遍elem,如果不设置n就一直输出

可以通过takewhile()直接截取一个有限的序列

it_a = itertools.count(1)
tw = itertools.takewhile(lambda x: x <= 10, it_a)
for i in tw:
    print(i, end=" ")
> 1 2 3 4 5 6 7 8 9 10 

一个用法

import itertools
dict_a = {'a': {'asd': {'a': [11, 22, 33]}, 'b': {'c': [12, 312]}}}
itea = iter(dict_a['a'].items())
for i in iter(lambda: tuple(itertools.islice(itea, 2)), ()):
    print(i)
>(('asd', {'a': [11, 22, 33]}), ('b', {'c': [12, 312]}))
相关标签: python iter