时间戳

首页 > 解决方案 > n 个列表中项目的所有可能组合

python - n 个列表中项目的所有可能组合

问题描述

我需要开发一个列表，其中包含 n 个列表中元素的所有可能组合。基本上我正在尝试找到所有可能的路径，稍后我将需要这些路径用于我的程序的另一部分。

我已经为两个列表做了一些简单的代码，但问题是我不知道用户会给出多少输入，所以我不得不猜测。目前我已经定义了一个输出所有可能组合的函数（只有一种方式，因为它们是路径）。我也一直在测试其他替代品，比如 itertools（我认为它可能是我问题的答案），或者使用 numpy 数组（问题是我的数组不是同质的）。

输入列表可能看起来像这样（3 维）：

chords = [[[1, 4, 8, 12], [1, 4, 10, 12]], [[4, 7, 13, 19], [4, 9, 13, 21]]]

我的函数可以生成两个列表之间的排列：

def combination(list1, list2):
    list = []
    for x in list1:
        for y in list2:
            list.append([x,y])
    return list

combination(chords[0], chords[1])

此功能按预期工作，但问题是例如当我引入时combination(combination(chords[0], chords[1]), chords[3])，它不单独计算chords[0]并且chords[1]（仍然按预期工作）。

编辑：

好的，就像@iBug 指出的那样，一个很好的方法是使用 itertools.product()：

bases_chords = [···] #It's a three dimensional array I've filled out  before
possibilities = [] #The list that will contain all the different combinations

for a in product(*bases_chords): #The asterisk means that I input everything on the list
    possibilities.append(a)

print(possibilities)
print(len(possibilities)) #Just to check if the dimensions are right

标签： pythonarrayslist

解决方案

itertools.product就是你要找的。它需要多个Iterables（列表是可迭代的）并生成一个生成器，该生成器循环遍历它们中的每一个的所有组合。

参见示例：

>>> for a, b, c in itertools.product([1, 2, 3], "abc", [True, False]):
...  print(a, b, c)
...
1 a True
1 a False
1 b True
1 b False
1 c True
1 c False
2 a True
2 a False
2 b True
2 b False
2 c True
2 c False
3 a True
3 a False
3 b True
3 b False
3 c True
3 c False
>>>

所以你的用例会变成：

itertools.product(*chords)

推荐阅读