嵌套循环迭代器
嵌套循环迭代器

嵌套循环迭代器

需求:

对于若干个节点,每个节点有若干种值,返回这些节点全部可能的值组合。
例如,a1=[1,2], a2=[5,6,7], 那么(a1,a2)可能的组合有2×3=6种,返回这6种组合。

实现:

使用python的内建函数itertools.product(*args, repeat=1)。该函数返回是可迭代输入对象的笛卡尔乘积。

代码片段示例如下,for i in itertools.product(*args)取出的i是一个个元组,每个元组是一种组合。

参考链接如下:

itertools — 为高效循环而创建迭代器的函数 — Python 3.9.1 文档

python 多层for循环嵌套的优化方法_J符离的博客-CSDN博客_python多层for嵌套循环

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注