python进阶系列 - 07 itertools 迭代器

最新推荐文章于 2025-02-23 21:23:51 发布

原创

最新推荐文章于 2025-02-23 21:23:51 发布 · 1k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python

本文介绍了Python标准库中的itertools模块，重点讲解了product()的笛卡尔积、permutations()的排列、combinations()的组合、accumulate()的累加器、groupby()的分组以及无限迭代器count(), cycle(), repeat()的功能和用法，并通过代码示例展示了它们的使用场景。" 126294165,11411117,GBase数据库表分区详解,"['数据库', '数据库设计', 'GBase']

在Python的标准库当中有这么一个神奇的工具库，它能让你使用最简单的方式写出更简洁高效的代码，这就是itertools，使用这个工具能为你生成一个优雅的迭代器。

这个模块提供了一些工具来处理迭代器。简单地说，迭代器是一种可以在 for 循环中使用的数据类型。 Python 中最常见的迭代器是列表。
可以在官网获得itertools的全部用法。

product() 笛卡尔积

这个工具计算输入迭代器的笛卡尔积。它等价于嵌套的 for 循环。例如，product(A, B) 返回((x,y) for x in A for y in B) 相同。

代码：

from itertools import product

prod = product([1, 2], [3, 4])
# 注意，我们将迭代器转换为列表以便打印
print(list(prod))
# 以允许迭代器的笛卡尔积，请指定重复次数
prod = product([1, 2], [3], repeat=2)
print(list(prod))

结果：

[(1, 3), (1, 4), (2, 3), (2, 4)]
[(1, 3, 1, 3), (1, 3, 2, 3), (2, 3, 1, 3), (2, 3, 2, 3)]

permutations() 排列

该函数返回迭代器中元素的所有可能排列（无重复元素）。

代码：

from itertools import permutations

perm = permutations([1, 2, 3])
print(list(perm))

# 可选参数：排列的长度。
perm = permutations([1, 2, 3],

最低0.47元/天解锁文章