元组的子元组

Question

我如何生成

[(0,), (1,), (2,), (0,1), (0,2), (1,2), (0,1,2)]

以编程的方式(也就是说，不用手工写出所有内容)？也就是说，元组(0,1,2)的所有非空子元组的列表。

(注意，这不是要求子集，而是子元组。)

Answer 1

>>> from itertools import combinations
>>> t = (0, 1, 2)
>>> print [subset for r in range(1,4) for subset in itertools.combinations(t,r)]
[(0,), (1,), (2,), (0, 1), (0, 2), (1, 2), (0, 1, 2)]

Itertools是一种强大的资源。你应该检查一下文档

Answer 2

您可以使用powerset() 食谱并删除空集：

from itertools import chain, combinations

def powerset(iterable):
    "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"
    s = list(iterable)
    return chain.from_iterable(combinations(s, r) for r in range(len(s)+1))

如下所示：

In [3]: [ss for ss in powerset([0,1,2]) if ss]
Out[3]: [(0,), (1,), (2,), (0, 1), (0, 2), (1, 2), (0, 1, 2)]

Answer 3

您想要的基本上是powerset，但是没有空集。通过修改python迭代工具页面中的菜谱，从1的组合大小开始

from itertools import combinations, chain

def almost_powerset(seq):
    return list(chain.from_iterable(combinations(seq, r) 
                for r in range(1, len(seq)+1)))

然后把你的顺序

lst = [0, 1, 2]
res = almost_powerset(lst)

这将生成大小1，然后大小2等的所有组合，直到序列的总长度，然后使用chain连接它们。