如何找到一系列向量的笛卡尔积？

Question

我有个问题，我现在想不起来。基本上，我的数据格式如下：

• 有一个向量有一定数量的元素（比如说，三个，所以让我们以(1, 2, 3)为例。在本例中，维度为3
• 此向量在列表中重复n次

在这个例子中，我有[(1, 2, 3), (1, 2, 3)]如果n = 2。我现在想要的是一个新的向量列表，其中列表中向量内的每个元素与其他元素中的每个其他元素相结合。在本例中，我的预期输出是：

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

现在，对于这样一个小的列表，可以使用简单的循环来完成，但是我的问题是：如何编写一个函数，为x维的输入向量和列表中y个这样的向量提供所有这样的组合？

Answer 1

你似乎需要itertools.product：

a = [(1,2,3), (1,2,3)]

list(product(*a))
#[(1, 1), (1, 2), (1, 3), (2, 1), (2, 2), (2, 3), (3, 1), (3, 2), (3, 3)]