python中的嵌套生成器函数

Question

考虑一个元组 v = (a,b,c) 和一个生成器函数 generate(x)，它从元组接收一个项目并为每个项目生成多个选项。

在元组中的每个项目上生成generate(x)结果的所有可能组合的pythonic方法是什么？

我可以这样做：

v = (a,b,c)
for d in generate(v[0]):
    for e in generate(v[1]):
        for f in generate(v[2]):
            print d,e,f

但这很丑，而且我需要一个通用的解决方案。

Answer 1

Python 2.6 的函数 itertools.product() 可以满足您的需求：

import itertools
v = (a, b, c)
for d, e, f in itertools.product(*(generate(x) for x in v)):
  print d, e, f

来自文档：

输入迭代的笛卡尔积。

等价于 a 中的嵌套 for 循环 生成器表达式。例如， product(A, B) 返回与 ((x,y) for x in A for y in B)。