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

发布时间：2020-12-20 11:52:33 所属栏目：Python 来源：网络整理

导读：我正在尝试创建组合.示例代码如下： a = [1,2,3],[1,4,5]combinations = list(itertools.product(*a)) 输出： [(1,1),(1,4),5),(2,(3,5)] 我不需要组合(1,1). 我已经尝试过以下代码： for comb in combinations: if comb[0] == comb[1]: combinations.remove

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

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

输出：

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

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

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

但是因为我必须在大数据上执行此操作.这花费了太多时间.

组合的元素也应该等于列表中的项目数.
例如：a = [1,[2,3,7],[4,5,1]
每个组合中的元素将是3,如(1,4)

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

解决方法

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

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

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

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

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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!