如何找到矢量列表的笛卡尔积?

时间:2018-04-18 11:57:17

标签: python list iterator permutation

我遇到了一个问题,目前无法解决这个问题。基本上,我的数据格式如下:

  • 有一个具有一定数量元素的Vector(例如,三个,所以我们以(1, 2, 3)为例。在这个例子中,维度为3。
  • 此向量在列表中重复n

所以我在此示例[(1, 2, 3), (1, 2, 3)]中有n = 2。我现在想要的是一个新的向量列表,其中列表中Vector内的每个元素与其他元素中的每个元素组合在一起。在这个例子中,我的预期输出是:

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

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

1 个答案:

答案 0 :(得分:2)

您似乎需要includes

itertools.product