python - 在 Python 中获取集合的子集

Question

假设我们需要编写一个函数来给出一个集合的所有子集的列表。下面给出了函数和doctest。而我们需要完成函数的整个定义

def subsets(s):
   """Return a list of the subsets of s.

   >>> subsets({True, False})
   [{False, True}, {False}, {True}, set()]
   >>> counts = {x for x in range(10)} # A set comprehension
   >>> subs = subsets(counts)
   >>> len(subs)
   1024
   >>> counts in subs
   True
   >>> len(counts)
   10
   """
   assert type(s) == set, str(s) + ' is not a set.'
   if not s:
       return [set()]
   element = s.pop() 
   rest = subsets(s)
   s.add(element)    

---

它必须不使用任何内置函数

我的方法是在rest中添加“element”，然后全部返回，但是我不太熟悉如何在Python中使用set、list。

Answer 1

查看powerset() itertools docs 中的食谱.

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))

def subsets(s):
    return map(set, powerset(s))