基本上,我想建立一个列表理解两个迭代器的"笛卡尔积".想想以下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中获得类似的行为吗?
你在问这个吗?
[ (i,j) for i in range(1,3) for j in range(1,5) ]
笛卡尔积在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)]
关于嵌套理解的有趣事实:它模仿嵌套的"for"循环,因此内部的可以使用外部的值.这在笛卡儿产品案例中没用,但很有用.例如:
[ (i,j) for i in range(10) for j in range(i) ]
产生所有对(i,j)
其中0>=i>j>10
.