python中的itertools与iter

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 – 为什么来自`from itertools import chain`工作但不是`import itertools.chain as chain`?


下一篇:python – 如何在使用itertools.tee检查下一个元素时最小化空间成本?