Python告别数据处理卡顿之itertools模块使用详解
概要
itertools模块是Python标准库中一个功能强大的工具模块，专门用于创建迭代器的函数集合。该模块提供了一系列高效的迭代器构建块，能够帮助开发者以更加优雅和高效的方式处理数据序列。对于需要处理大量数据或者需要节省内存的应用场景，itertools模块显得尤为重要。
无限迭代器
1、count函数的高效使用
count函数是itertools模块中最基础的无限迭代器之一，它能够生成从指定起始值开始的连续数字序列。与传统的for循环相比，count函数提供了更加灵活的计数方式，支持自定义起始值和步长。
import itertools# 创建一个从10开始，步长为2的计数器
counter = itertools.count(10, 2)# 获取前5个值
result = [next(counter) for _ in range(5)]
print(f"计数结果: {result}")# 实际应用：生成产品编号
def generate_product_ids(prefix="PROD", start=1000):
counter = itertools.count(start)
whileTrue:
yieldf"{prefix}{next(counter):04d}"# 生成5个产品编号
product_generator = generate_product_ids()
products = [next(product_generator) for _ in range(5)]
print(f"产品编号: {products}")
运行结果：
计数结果: [10, 12, 14, 16, 18]
产品编号: ['PROD1000', 'PROD1001', 'PROD1002', 'PROD1003', 'PROD1004']
2、cycle函数的循环处理
cycle函数能够创建一个无限循环的迭代器，它会不断重复输入序列中的元素。这个函数在需要循环使用某些固定值的场景中非常有用，比如轮询处理、状态切换或者资源分配等应用。
import itertools# 创建一个循环迭代器
colors = itertools.cycle(['红', '绿', '蓝'])# 模拟交通信号灯循环
def traffic_light_simulation(duration=10):
light_cycle = itertools.cycle(['红灯', '绿灯', '黄灯'])
for second in range(duration):
current_light = next(light_cycle)
print(