如何从一个列表中生成所有可能的球员球队？

Question

我有一个名单，比如说，3 名玩家：

[1, 2, 3]

如何在 python 中生成列表列表，格式为：

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

代表所有可以与上述球员组成的球队？

Answer 1

您可以使用itertools.combinations()，我们可以将r 参数设置为从1 到列表长度(x) 的所有长度，以获取所有可能在列表理解中被展平的组合。

from itertools import combinations

x = [1, 2, 3]
result = [c for i in range(1, len(x)+1) for c in combinations(x, i)]
print(result)  # -> [(1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3)]

Answer 2

使用https://docs.python.org/3/library/itertools.html#itertools.combinations

它完全符合您的要求。

import itertools
players = [1, 2, 3]
print(list(itertools.chain.from_iterable(itertools.combinations(players, r) for r in range(1, len(players) + 1))))

输出：

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

由于使用了 itertools.chain，这可能是最有效的答案

Answer 3

您可以使用具有给定尺寸的itertools.combinations 来生成固定尺寸组合。为了概括，您可以在所有尺寸上使用 for 循环。代码如下所示：

import itertools

my_list = [1, 2, 3]
for L in range(0, len(my_list)+1):
    for subset in itertools.combinations(my_list, L):
        print(subset)

Answer 4

Itertools 是这些操作的朋友：

import itertools

l = [1, 2, 3]

l = [list(x) for y in range(1,len(l)+1) for x in itertools.combinations(l,y) ]
print(l)

给予：

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

Answer 5

您可以使用itertools。您要做的是生成给定列表的 powerset。

>>> import itertools
>>> a=[1,2,3]
>>> out=[]
>>> for i in range(len(a)+1):
    out+=list(itertools.combinations(a,i))


>>> out
[(), (1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3)]
>>> 

---

您可以编写recursive函数来生成powerset，如下所示：

>>> def powerset(s,idx,curr,out):
    if idx==len(s):
        out.append(curr)
        return
    (powerset(s,idx+1,curr+[s[idx]],out))
    (powerset(s,idx+1,curr,out))
    return sorted(out,key=lambda x:len(x))

>>> z=powerset(a,0,[],[])
>>> z
[[], [1], [2], [3], [1, 2], [1, 3], [2, 3], [1, 2, 3]]