缺少 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), [])