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