如何将 itertools.product 应用于列表列表的元素？

Question

我有一个数组列表，我想获得数组中元素的笛卡尔积。

我会用一个例子来使这个更具体...

itertools.product 似乎可以解决问题，但我被困在一个小细节上。

arrays = [(-1,+1), (-2,+2), (-3,+3)];

如果我这样做

cp = list(itertools.product(arrays));

我明白了

cp = cp0 = [((-1, 1),), ((-2, 2),), ((-3, 3),)]

但我想得到的是

cp1 = [(-1,-2,-3), (-1,-2,+3), (-1,+2,-3), (-1,+2,+3), ..., (+1,+2,-3), (+1,+2,+3)].

我尝试了一些不同的方法：

cp = list(itertools.product(itertools.islice(arrays, len(arrays))));
cp = list(itertools.product(iter(arrays, len(arrays))));

他们都给了我cp0而不是cp1。

有什么想法吗？

提前致谢。

Answer 1

>>> list(itertools.product(*arrays))
[(-1, -2, -3), (-1, -2, 3), (-1, 2, -3), (-1, 2, 3), (1, -2, -3), (1, -2, 3), (1, 2, -3), (1, 2, 3)]

这会将所有对作为单独的参数提供给product，然后它将为您提供它们的笛卡尔积。

您的版本不起作用的原因是您只给了product 一个参数。请求一个列表的笛卡尔积是一种简单的情况，并返回一个仅包含一个元素的列表（作为参数给出的列表）。

Answer 2

>>> arrays = [(-1,+1), (-2,+2), (-3,+3)]
>>> list(itertools.product(*arrays))
[(-1, -2, -3), (-1, -2, 3), (-1, 2, -3), (-1, 2, 3), (1, -2, -3), (1, -2, 3), (1, 2, -3), (1, 2, 3)]

Answer 3

您可以使用 itertools.product 在三个 rurch 中完成此操作

lst=[]
arrays = [(-1,+1), (-2,+2), (-3,+3)]  

import itertools 

for i in itertools.product(*arrays):
         lst.append(i)



print(lst)