给定列表的所有可能的列表组合

Question

我想要给定大小的列表的所有可能组合。 例如，我有一个给定的列表 K，例如K = [3, 5, 2] 。 使用以下代码，我得到了我想要的。 但是，我如何将它推广到给定大小的任何列表，而不仅仅是添加 for 循环？

assignment= []
for l in range(K[0]):
    for m in range(K[1]):
        for n in range(K[2]):
            a = [l,m,n]
            assignment.append(a)

Answer 1

itertools库通常可以帮助解决此类问题。 特别是itertools.product() ，使用*向其传递范围列表或迭代器：

list(itertools.product(*(range(n) for n in K))

或者

list(itertools.product(*map(range, K)))