通过组合元组元素，获取列表元组的乘积？

Question

我有一个元组列表，想通过组合各个元组元素来获取它们的乘积。

例如：

lists = [
    [(1,), (2,), (3,)],
    [(4,), (5,), (6,)]
]
p = itertools.product(*lists)
for product in p:
    print product

这会导致一堆元组的元组：

((1,), (4,))
((1,), (5,))
((1,), (6,))
((2,), (4,))
((2,), (5,))
((2,), (6,))
((3,), (4,))
((3,), (5,))
((3,), (6,))

我想要的是像这样的元组列表：

(1,4)
(1,5)
(1,6)
(2,4)
(2,5)
(2,6)
(3,4)
(3,5)
(3,6)

我也希望您能将其应用于任何数量的元组列表。

因此，在三个元组的情况下：

lists = [
    [(1,), (2,), (3,)],
    [(4,), (5,), (6,)],
    [(7,), (8,), (9,)]
]
p = itertools.product(*lists)
for product in p:
    print product

I'd like:

(1, 4, 7)
(1, 4, 8)
(1, 4, 9)
(1, 5, 7)
(1, 5, 8)
(1, 5, 9)
(1, 6, 7)
(1, 6, 8)
(1, 6, 9)
(2, 4, 7)
(2, 4, 8)
(2, 4, 9)
(2, 5, 7)
(2, 5, 8)
(2, 5, 9)
(2, 6, 7)
(2, 6, 8)
(2, 6, 9)
(3, 4, 7)
(3, 4, 8)
(3, 4, 9)
(3, 5, 7)
(3, 5, 8)
(3, 5, 9)
(3, 6, 7)
(3, 6, 8)
(3, 6, 9)

- FGreg

Answer 1

您可以简单地使用itertools.chain.from_iterable来展开内部元组，就像这样：

for product in p:
    print tuple(itertools.chain.from_iterable(product))

例如，

>>> from itertools import chain, product
>>> [tuple(chain.from_iterable(prod)) for prod in product(*lists)]
[(1, 4), (1, 5), (1, 6), (2, 4), (2, 5), (2, 6), (3, 4), (3, 5), (3, 6)]

- thefourtheye

---

这适用于 n 大小的列表吗？ - PyNEwbie

@PyNEwbie 是的，它确实可以。 - thefourtheye

Answer 2

你也可以使用 map 或 imap 与 sum 或 @thefourtheye 的解决方案来获得所需的结果。

from itertools import product, chain, imap
p1 = imap(lambda x:sum(x,tuple()),product(*lists))
p2 = imap(lambda x:tuple(chain.from_iterable(x)),product(*lists))

- Copperfield

Answer 3

正如其他人所提到的，最好先展平列表

lists = [[x[0] for x in tup] for tup in lists] #flatten
from itertools import product
for p in product(*lists):
    print p

那个带有“flatten”注释的第一行代码将把你的列表对象转换成类似于[ [1,2,3] , [4,5,6]]的形式，然后你就可以像以前一样使用itertools product函数了。

- Garrett R