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

2019年8月3日 198次阅读

我正在尝试创建组合.示例代码如下：

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)

请建议一种避免这种组合的方法.

最佳答案 对于两个迭代,一个简单的列表理解会做：

>>> 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)]

如果您需要过滤任意数量的iterables的产品,那么最好使用sets来检查组合的所有元素是否都是不同的：

>>> 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)]

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

点赞