itertools
是 Python 内建的一个高效处理迭代器的模块,提供了创建复杂迭代器的函数工具。它包含一系列用于迭代、组合、排列、过滤等功能的迭代器构建工具,常用于数据处理和算法设计。下面是 itertools
模块中一些常见的函数介绍:
1. 无限迭代器
-
itertools.count(start=0, step=1)
:- 生成从
start
开始,每次增加step
的无限序列。常用于构建自增序列。
- 生成从
import itertools
for i in itertools.count(10, 2):
print(i)
if i > 20:
break
itertools.cycle(iterable)
:
- 无限循环遍历可迭代对象。
import itertools
cycle_iter = itertools.cycle('ABC')
for _ in range(6):
print(next(cycle_iter))
itertools.repeat(object, times=None)
:
- 无限次或指定次数重复给定对象。
import itertools
repeat_iter = itertools.repeat('hello', 3)
print(list(repeat_iter))