博客

Python的itertools模块是一个包含了很多用于迭代器操作的工具函数的模块。该模块提供了一些效率很高的操作序列的函数，可以用于快速生成和操作更有趣和复杂的序列。

下面是itertools模块中一些常用的函数以及示例：

itertools.count()函数

itertools.count()函数会从一个数开始，不断地给这个数加1，生成一个无限的整数迭代器。

import itertools

# 从-3开始生成无限长的整数序列
a = itertools.count(-3)

# 打印前10个数
for i in range(10):
    print(next(a))

输出结果：

-3
-2
-1
0
1
2
3
4
5
6

itertools.repeat()函数

itertools.repeat()函数用于生成重复的值。

import itertools

# 常量迭代器，不停返回'hello, world!'
a = itertools.repeat('hello, world!', 5)

# 打印5个'hello, world!'
for i in range(5):
    print(next(a))

输出结果：

hello, world!
hello, world!
hello, world!
hello, world!
hello, world!

通过itertools模块的这些函数，可以快速地生成和操作序列，大大提高编程效率。除了本文介绍的函数，itertools模块还包含了许多其他有用的函数，读者可自行查阅文档学习。