Python - 没有重复的列表列表的所有组合

Question

我想在组合结果中组合字符串列表列表而不重复字符串，但我不知道如何做到这一点。

例如：

# Input
pickup_list = [["a", "b", "c"], ["d", "e"], ["a", "c", "f"]]
print(combinaison_function(pickup_list))

# Output
> [["a", "d", "c"], ["a", "d", "f"], ["a", "e", "c"], ["a", "e", "f"], ["b", "d", "a"], ["b", "d", "c"], ...]

在示例中 ["a", "d", "a"] 没有返回，因为 "a" 是重复的并且我想做。

我想解决方案将使用 itertools function 但我不知道该怎么做。

感谢您提前回复。

Answer 1

这可以通过itertools.product()然后过滤重复来完成。 检查重复项的一种快速方法是强制转换为set 。

comb_list = []
for comb in itertools.product(*pickup_list):
    if len(set(comb))==len(comb):
        comb_list.append(comb)

使用您的pickup_list ，我得到：

[('a', 'd', 'c'), ('a', 'd', 'f'), ('a', 'e', 'c'), ('a', 'e', 'f'), ('b', 'd', 'a'), ('b', 'd', 'c'), ('b', 'd', 'f'), ('b', 'e', 'a'), ('b', 'e', 'c'), ('b', 'e', 'f'), ('c', 'd', 'a'), ('c', 'd', 'f'), ('c', 'e', 'a'), ('c', 'e', 'f')]

Answer 2

您可以制作自己的递归生成器：

def uCombo(L,used=None):
    if used is None: used = set()   # track used values
    if not L: yield [];return       # end of recursion
    for n in L[0]:                  # combine each unused value
        if n in used: continue      # with remaining sub-lists
        yield from ( [n]+c for c in uCombo(L[1:],used | {n}) )
    

output：

pickup_list = [["a", "b", "c"], ["d", "e"], ["a", "c", "f"]]
for combo in uCombo(pickup_list): print(combo)
                     
['a', 'd', 'c']
['a', 'd', 'f']
['a', 'e', 'c']
['a', 'e', 'f']
['b', 'd', 'a']
['b', 'd', 'c']
['b', 'd', 'f']
['b', 'e', 'a']
['b', 'e', 'c']
['b', 'e', 'f']
['c', 'd', 'a']
['c', 'd', 'f']
['c', 'e', 'a']
['c', 'e', 'f']