我正在尝试创建组合。示例代码如下:
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)
请建议一种避免这种组合的方法。
(1, 1, 1)
,还是也包括(1, 1, 2)
和(1, 2, 1)
? - poke