python - itertools 产品不应包含具有重复值的组合

Question

我正在尝试创建组合。示例代码如下:

a = [1, 2, 3], [1, 4, 5]
combinations = list(itertools.product(*a))

输出:

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

我不需要组合 (1,1)。我已经尝试过以下代码:

for comb in combinations:
    if comb[0] == comb[1]:
        combinations.remove(comb)

但是因为我必须对大数据执行此操作。这需要太多时间。

此外，组合中的元素应等于列表中的项目数。例如:a = [1,2,3], [2,3,7],[4,5,1]每个组合中的元素都是 3，如 (1,2,4)

请提出避免此类组合的方法。

Answer 1

对于两个可迭代对象，一个简单的列表理解就可以:

>>> from itertools import product
>>> a = [1, 2, 3], [1, 4, 5]
>>> [(x, y) for x, y in product(*a) if x != y]
[(1, 4), (1, 5), (2, 1), (2, 4), (2, 5), (3, 1), (3, 4), (3, 5)]

如果您需要过滤任意数量的可迭代对象的乘积，那么最好使用集合来检查组合的所有元素是否不同:

>>> a = [1, 2, 3], [1, 4, 5], [1, 8, 9]
>>> [p for p in product(*a) if len(set(p)) == len(p)]
[(1, 4, 8), (1, 4, 9), (1, 5, 8), (1, 5, 9), (2, 1, 8), (2, 1, 9), (2, 4, 1), (2, 4, 8), (2, 4, 9), (2, 5, 1), (2, 5, 8), (2, 5, 9), (3, 1, 8), (3, 1, 9), (3, 4, 1), (3, 4, 8), (3, 4, 9), (3, 5, 1), (3, 5, 8), (3, 5, 9)]

顺便说一句，永远不要改变你正在循环的列表，因为这很可能会产生一个不正确的循环。