如何生成包含列表首元素的升序幂集并筛选含该元素的子集

针对你的需求，这里有两种简单的实现方式：

方法1：筛选全幂集结果 #

原powerset函数生成的所有子集中，只要保留包含元素1的那些就行。因为combinations生成的子集严格遵循原列表顺序，天然是升序排列，直接过滤即可：

from itertools import chain, combinations

backlog = [1, 2, 3, 4, 5]

def powerset(backlog):
    s = backlog
    return chain.from_iterable(combinations(s, r) for r in range(len(s) + 1))

# 筛选出包含1的子集
target_subsets = [subset for subset in powerset(backlog) if 1 in subset]

得到的target_subsets就是从(1)、(1,2)、(1,3)一直到(1,2,3,4,5)的所有符合要求的子集，顺序完全符合预期。

方法2：直接生成目标子集（更高效） #

如果原列表元素较多，先生成全幂集再筛选会浪费资源，不如直接构造包含1的子集：把1和剩余元素（[2,3,4,5]）的所有子集（包括空集）组合起来，直接得到所有包含1的升序子集：

from itertools import chain, combinations

backlog = [1, 2, 3, 4, 5]
first_elem = backlog[0]
remaining_elements = backlog[1:]

# 生成剩余元素的所有子集（包括空集）
remaining_subsets = chain.from_iterable(combinations(remaining_elements, r) for r in range(len(remaining_elements) + 1))

# 将1与每个剩余子集拼接，得到目标结果
target_subsets = [(first_elem,) + subset for subset in remaining_subsets]

这种方式跳过了所有不含1的子集生成，性能更优，最终结果和方法1完全一致。

内容的提问来源于stack exchange，提问作者Mathijsvdk