python:通过集合的子集为循环创建迭代器

Question

我有一个连续整数序列，例如 [1, 2, 3]。我想创建一个遍历该集合所有子集的迭代器。在这种情况下 [], [1],...,[1,2,3]。我怎么能这样做？

Answer 1

itertools documentation包含此构造的配方，名称为 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))