小编pan*_*anr的帖子

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

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)

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