使用 itertools 枚举
笛卡尔积、组合与排列
使用 itertools 枚举 是 CoddyKit 上的免费 Coding Interview Prep 课时。 这是第 2 节课,共 4 节。 你可以在下方免费阅读本课时的完整内容 — 然后在浏览器中使用内置代码编辑器和全天候 AI 导师进行实践。 这是 Coding Interview Prep 学习路径的一部分,你的进度在网页和 CoddyKit 应用中同步。 Coding Interview Prep 课程共包含 4 节课。
让迭代工具库处理循环
Python 的迭代工具库会为您生成组合和排列,因此您可以少写循环代码,也能减少错误。🙂
先导入迭代工具库
所有功能都位于同一个模块中,因此请先导入迭代工具库。然后您就可以直接调用 product、permutations 和 combinations。
import itertools用 product 替代嵌套循环
使用 product 可以替代针对相同选项的多层嵌套循环。它会生成给定范围内所有选取结果组成的元组。
for combo in itertools.product(range(3), repeat=2):
print(combo)用 permutations 生成排列
当顺序很重要时,permutations会列出元素的每种排列。它非常适合处理小规模、类似旅行商问题的暴力搜索。
for p in itertools.permutations([1, 2, 3]):
print(p)用 combinations 进行选择
当顺序不重要时,combinations会选出每个大小为 r 的组合。它非常适合选择一组元素进行测试。
for c in itertools.combinations([1, 2, 3, 4], 2):
print(c)排列与组合的区别
关键区别在于顺序:permutations 会将 (1,2) 和 (2,1) 视为不同结果,而 combinations 会将它们视为同一种选择。
注意增长速度
排列数量按 n! 增长,速度会迅速膨胀。即使 n 等于 11,也已经超过四千万种排列,因此请让 n 保持较小。
它们采用惰性生成
这些函数是生成器:它们一次生成一个元素,而不是构建一个巨大的列表,因此循环时占用的内存很少。
结合检查条件
先生成候选项,然后只保留通过测试的候选项。这种模式可以将困难的搜索变成简短易读的循环。
best = min(itertools.permutations(cities), key=tour_length)允许重复的组合
需要选择可以重复的元素吗?请使用允许重复的组合功能,让同一个元素可以在一个组合中出现多次。
选择正确的工具
请问自己两个问题:顺序重要吗?元素可以重复吗?答案会直接指向正确的迭代工具库函数。
快速检查
您必须枚举从 6 个元素中选择 3 个元素的所有方式,并且顺序不重要。
回顾
使用 product、permutations 和 combinations 可以清晰地完成枚举。请根据顺序是否重要以及元素是否重复来进行选择,同时注意数量增长。🚀
常见问题解答
「使用 itertools 枚举」课时是免费的吗?
是的 — 「使用 itertools 枚举」的完整文本可在网页上免费阅读。要进行交互式练习(内置代码编辑器和全天候 AI 导师)并解锁 Coding Interview Prep 课程的其余内容,请升级到 CoddyKit PRO。 Coding Interview Prep 课程共包含 4 节课。
「使用 itertools 枚举」这节课中我会学到什么?
笛卡尔积、组合与排列 你通过在浏览器中直接运行的动手代码来练习 Coding Interview Prep,全天候 AI 导师会在你学习这节课的过程中回答你的问题。
学习 Coding Interview Prep 需要有经验吗?
无需任何先前经验。CoddyKit 上的 Coding Interview Prep 课程适合初学者到高级学习者,你可以从这里开始或从头开始,按照自己的节奏学习。 这是第 2 节课,共 4 节。
「使用 itertools 枚举」课时需要多长时间?
大多数 CoddyKit 课程大约需要 5–10 分钟。每节课都很精短且互动,所以你能稳步进步,并在网页和应用中从离开的地方继续。
我能在这节 Coding Interview Prep 课中编写并运行代码吗?
能。每节 Coding Interview Prep 课都包含内置代码编辑器,你可以在浏览器中直接编写并运行真实代码,并获得即时 AI 反馈 — 无需本地设置。
此课程中的所有课时
- 暴力搜索也是有效策略
- 使用 itertools 枚举
- 使用位掩码枚举子集
- 巧妙缩小搜索空间