Python 从列表列表中找到所有不同的选择

Question

我想从 python 的列表中获取所有不同的选择。 例如，拿一个像这样的列表，

list=[[a,b,c],[a,d,e]]

我只想从每个列表中的一个中选择 select 并获得所有可能的不同选择。 所以上面列出的可能选择看起来像，

    [a,a]
    [a,d]
    [a,e]
    [b,a]
    [b,d]
    [b,e]
    [c,a]
    [c,d]
    [c,e]

不使用任何外部库，我该怎么做？

Answer 1

使用递归生成器最容易实现。

def yield_combos(lst):
    if lst:
        for el in lst[0]:
            for combo in yield_combos(lst[1:]):
                yield [el] + combo
    else:
        yield []


lst = [['a','b','c'], ['a','d','e']]
for combo in yield_combos(lst):
    print(combo)

如果您更喜欢 output 作为列表，您可以随时将其转换为一个：

print(list(yield_combos(lst)))

小提示：

• 我在问题使用list的地方调用了变量lst ； 覆盖内置函数不是一个好主意，在这种情况下，这样做会阻止对list的调用。

• 为了一个独立的程序，我在调用者中设置了一些字符串值（问题有a代替'a'等）。

Answer 2

你可以使用生成器，

perm = [(i,j) for i in list[0] for j in list[1]]

此外，如果您可以将其转换为生成器 object，如果您担心 memory，

perm = ((i,j) for i in list[0] for j in list[1]])

有人告诉我你有 n 个列表并且想要排列所有列表，你可以使用 while 循环

while(len(list)>1):
    a = list.pop()
    b = list.pop()
    list.append([(i, j) for i in a for j in b])

或制作递归 function。 请注意，答案将是 list[0] 而不是列表本身。

Answer 3

最简单的方法是使用两个函数

def funcA(x, arr):
  return [(x,c) for c in arr]

def funcB(arr1, arr2):
  ans = []
  for x in arr1:
    ans.extend(funcA(x, arr2))
  return ans

以任何需要的方式调用它：

funcB(your_list[0], your_list[1])

Answer 4

最 Pythonic 的方式是使用 itertools，您可以简单地使用 itertools 库中的产品，如下所示：

my_list = [[a,b,c],[a,d,e]]
# will contain: [(a,a), (a,d)...]
all_combinations = list(itertools.product(*my_list))

“*”将列表解包到它包含的列表中，就像这样写：

# will contain: [(a,a), (a,d)...]
list(itertools.product([a,b,c],[a,d,e]))

值得一提的是，这适用于任意数量的子列表。

my_list = [[a,b,c],[a,d,e], [1,2,3]]

# will contain: [(a,a,1), (a,a,2)...]
all_combinations = list(itertools.product(*my_list))

希望有帮助

Answer 5

l=[['a','b','c'],['a','d','e']]
def combinations(L, tmp=None):
    if tmp is None:
        tmp = []
    if L==[]:
        print (tmp)
    else:
        for i in L[0]:
            product(L[1:], tmp+[i])
combinations(l)

这有帮助吗？