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]}))