缺少numpy数组的循环置换

Question

我有numpy数组，每个数组包含2个浮点数：

Unit_1 = [40000.0, 47000.0]
Unit_2 = [12000.0, 14000.0]
Unit_3 = [16000.0, 18000.0]

我想根据单位的值为单位组合构建排列，例如：

[(40000, 12000, 16000), (40000, 12000, 18000), (40000, 14000, 160000), (40000, 14000, 18000).........]

获取所有可能的排列。

我正在使用以下内容：

list(list(zip(r, p, q)) for (r, p,q) in zip(repeat(Unit_1), permutations(Unit_2), permutations(Unit_3)))

输出为：

[[(40000.0, 12000.0, 16000.0), (47000.0, 14000.0, 18000.0)], [(40000.0, 14000.0, 18000.0), (47000.0, 12000.0, 16000.0)]]

如何获得剩余的排列？

Answer 1

您似乎不需要排列，但是3个列表中的笛卡尔乘积。使用itertools.product。

items = list(itertools.product(Unit_1, Unit_2, Unit_3))

要生成排列，您需要上述集合中的每个项目的排列。因此，类似

perms = sum((list(permutations(item)) for item in items), [])