在Python中,是否有一种使用多个迭代器的列表理解的简明方法？

Question

基本上,我想建立一个列表理解两个迭代器的"笛卡尔积".想想以下Haskell代码:

[(i,j) | i <- [1,2], j <- [1..4]]

产量

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

我能以简洁的方式在Python中获得类似的行为吗？

Answer 1

你在问这个吗？

[ (i,j) for i in range(1,3) for j in range(1,5) ]

Answer 2

笛卡尔积在itertools模块中(在2.6中).

>>> import itertools
>>> list(itertools.product(range(1, 3), range(1, 5)))
[(1, 1), (1, 2), (1, 3), (1, 4), (2, 1), (2, 2), (2, 3), (2, 4)]

Answer 3

关于嵌套理解的有趣事实:它模仿嵌套的"for"循环,因此内部的可以使用外部的值.这在笛卡儿产品案例中没用,但很有用.例如:

[ (i,j) for i in range(10) for j in range(i) ] 

产生所有对(i,j)其中0>=i>j>10.