如何用Python编写返回列表所有可能子集的函数？

嘿，我来帮你搞定生成列表子集/子列表的Python代码问题！先明确两个容易混淆的概念，再给你具体实现方案：

这里的子集指原列表元素的任意组合，不要求元素连续，包括空集和原列表本身，总数是 2^n（n为列表长度）。

方法1：用itertools.combinations（简洁高效） #

Python标准库的itertools模块提供了现成的组合生成工具，我们可以遍历所有可能的子集长度（从0到列表长度），生成对应长度的所有组合，再合并成结果：

import itertools

def get_all_subsets(input_list):
    subsets = []
    # 遍历所有可能的子集长度：0（空集）到列表总长度
    for subset_length in range(len(input_list) + 1):
        # 生成当前长度的所有元素组合，转成列表加入结果
        for combo in itertools.combinations(input_list, subset_length):
            subsets.append(list(combo))
    return subsets

# 测试示例
my_list = [1, 2, 3]
print(get_all_subsets(my_list))
# 输出：[[], [1], [2], [3], [1, 2], [1, 3], [2, 3], [1, 2, 3]]

方法2：递归实现（适合理解原理） #

如果想手动实现核心逻辑，递归是个好思路：每次取出列表的第一个元素，递归处理剩余元素的子集，再把第一个元素加到每个剩余子集里，最后合并两部分结果：

def get_all_subsets_recursive(input_list):
    # 基线条件：空列表的子集只有空列表
    if not input_list:
        return [[]]
    # 取出第一个元素
    first_element = input_list[0]
    # 递归获取剩余元素的所有子集
    rest_subsets = get_all_subsets_recursive(input_list[1:])
    # 合并：原剩余子集 + 加入第一个元素后的新子集
    return rest_subsets + [[first_element] + subset for subset in rest_subsets]

# 测试示例
print(get_all_subsets_recursive([1, 2, 3]))
# 输出和上面完全一致

如果你说的子列表是指原列表中连续元素组成的列表（比如[1,2,3]的连续子列表包括[1]、[1,2]、[2]等），可以用双重循环遍历所有起始和结束索引：

def get_all_contiguous_sublists(input_list):
    sublists = []
    list_length = len(input_list)
    # 遍历所有起始位置
    for start_idx in range(list_length):
        # 遍历所有结束位置（从起始位置到列表末尾）
        for end_idx in range(start_idx + 1, list_length + 1):
            # 切片获取连续子列表
            sublists.append(input_list[start_idx:end_idx])
    # 可选：如果需要包含空集，就加上这一行
    sublists.append([])
    return sublists

# 测试示例
print(get_all_contiguous_sublists([1, 2, 3]))
# 输出：[[1], [1, 2], [1, 2, 3], [2], [2, 3], [3], []]

如果输入列表有重复元素（比如[1,1,2]），上面的方法会生成重复的子集。要得到去重后的结果，可以用集合先去重（注意集合不能存列表，所以先转成元组）：

import itertools

def get_unique_subsets(input_list):
    unique_subsets = set()
    for subset_length in range(len(input_list) + 1):
        for combo in itertools.combinations(input_list, subset_length):
            unique_subsets.add(combo)
    # 把元组转成列表返回
    return [list(s) for s in unique_subsets]

# 测试示例
print(get_unique_subsets([1,1,2]))
# 输出：[[], [1], [2], [1, 1], [1, 2], [1, 1, 2]]

如果还有其他需求，比如优化大列表的性能、生成特定长度的子集等，随时告诉我！

内容的提问来源于stack exchange，提问作者Ibrahim Isah Nayaya