获取一系列列表的笛卡尔积？

Question

如何从一组列表中获取笛卡尔积（每种可能的值组合）？

输入：

somelists = [
   [1, 2, 3],
   ['a', 'b'],
   [4, 5]]

期望的输出：

[(1, 'a', 4), (1, 'a', 5), (1, 'b', 4), (1, 'b', 5), (2, 'a', 4), (2, 'a', 5) ...]




Answer 1

import itertools>>> for i in itertools.product([1,2,3],['a','b'],[4,5]):...         print i...(1, 'a', 4)(1, 'a', 5)(1, 'b', 4)(1, 'b', 5)(2, 'a', 4)(2, 'a', 5)(2, 'b', 4)(2, 'b', 5)(3, 'a', 4)(3, 'a', 5)(3, 'b', 4)(3, 'b', 5)>>>

Answer 2

对于Python 2.5及更早版本：>>> [(a, b, c) for a in [1,2,3] for b in ['a','b'] for c in [4,5]][(1, 'a', 4), (1, 'a', 5), (1, 'b', 4), (1, 'b', 5), (2, 'a', 4),  (2, 'a', 5), (2, 'b', 4), (2, 'b', 5), (3, 'a', 4), (3, 'a', 5),  (3, 'b', 4), (3, 'b', 5)]这是一个递归版本product()（只是一个插图）：def product(*args):    if not args:        return iter(((),)) # yield tuple()    return (items + (item,)             for items in product(*args[:-1]) for item in args[-1])例：>>> list(product([1,2,3], ['a','b'], [4,5])) [(1, 'a', 4), (1, 'a', 5), (1, 'b', 4), (1, 'b', 5), (2, 'a', 4),  (2, 'a', 5), (2, 'b', 4), (2, 'b', 5), (3, 'a', 4), (3, 'a', 5),  (3, 'b', 4), (3, 'b', 5)]>>> list(product([1,2,3]))[(1,), (2,), (3,)]>>> list(product([]))[]>>> list(product())[()]