从python中的不同列表中生成所有可能的对

Question

我用这种方式进行所有组合：

import itertools

lst = [[1, 2, 3], [1,2,2,4]]
combs = []

for i in xrange(1, len(lst)+1):
    combs.append(i)
    els = [list(x) for x in itertools.combinations(lst, i)]
    combs.append(els)

但我想要的是每个列表都包含里面所有可能的元素组合。使用上面的解决方案，每个元素对都是分散的。我该如何解决？

Answer 1

你在寻找这样的东西吗？

import itertools

lst = [1, 2, 3], [1, 2, 2, 4]
combs = []

for i in range(len(lst)):
    els = [list(x) for x in itertools.combinations(lst[i], 2)]
    combs.append(els)

print(combs)

输出

[[[1, 2], [1, 3], [2, 3]], [[1, 2], [1, 2], [1, 4], [2, 2], [2, 4], [2, 4]]]

既然你说，你想要-each element in its own list make pairs and put back in that list，所以我假设[1, 2, 3] 应该转换为[[1, 2], [1, 3], [2, 3]]。上面的例子做同样的事情！

---

更新

如果您想从这些列表中生成所有可能的组合（长度大于 1），那么您可以执行以下操作。

import itertools

lst = [1, 2, 3], [1, 2, 2, 4]
combs = []

for i in range(len(lst)):
    temp_list = []
    for j in range(len(lst[i])+1):
        if j < 2: # skipping zero and one length combinations
            continue
        els = [x for x in itertools.combinations(lst[i], j)]
        temp_list.extend(els)

    # remove duplicate combinations
    new_list = []
    [new_list.append(i) for i in temp_list if not new_list.count(i)]
    combs.append(new_list)

print(combs)

输出

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