Python itertools - 仅创建所有可能产品的一个子集

Question

我正在编写一个脚本，该脚本使用 itertools 生成给定参数的所需组合。但是，当链接某些变量时，即排除某些组合时，我无法生成适当的“集合”。考虑以下：

import itertools

A = ['a1','a2','a3']
B = ['b1','b2','b3']
C = ['c1','c2']

如果我想生成这些元素的所有可能组合，我可以简单地使用 itertools.product()

all_combinations = list(itertools.product(A,B,C))

这给出了预期的

[('a1', 'b1', 'c1'), ('a1', 'b1', 'c2'), ('a1', 'b2', 'c1'), ... 
('a3', 'b2', 'c2'), ('a3', 'b3', 'c1'), ('a3', 'b3', 'c2')]

18种组合（3*3*2）

但是，我如何“链接”参数 A 和 B，使得每个返回的集合只包含 'an','bn' 元素？也就是说，我已经尝试过：

ABprecombine = zip(A,B)
limited_combinations = list(itertools.product(ABprecombine,C))

哪个返回

[(('a1', 'b1'), 'c1'), (('a1', 'b1'), 'c2'), (('a2', 'b2'), 'c1'),
 (('a2', 'b2'), 'c2'), (('a3', 'b3'), 'c1'), (('a3', 'b3'), 'c2')]

这是六个 (3*1*2) 所需的产品，但显然由于我创建它的方式，我现在有一个额外的元组。

当然，我可以生成所有组合，然后过滤掉给定的组合，但是有没有一种像上面那样“链接”参数的智能方法？

Answer 1

在这里，拉链A是B正确的选择。如果您愿意，您可以很容易地展平元组：

limited_combinations = [(a, b, c) for ((a, b), c) in itertools.product(zip(A, B), C)]

如果您想要更详细地控制生成的组合，事情会迅速变得更加复杂，甚至需要解决布尔可满足性等 NP 难题。如果发生这种情况，请查看现有的库以查找此类内容。