如何生成长度为n、元素为0、1或None的列表的所有可能组合？

代码错误原因 #

permutations函数的作用是对输入列表的元素做全排列，只能得到输入元素的重排结果，无法实现「每个位置独立从0、1、None三个值中任选」的需求，所以输出结果不符合预期。

正确实现方法 #

使用Python标准库的itertools.product生成笛卡尔积即可，它支持让每个位置独立从可选值中选择，刚好匹配需求场景。

完整可运行代码 #

from itertools import product

def generate_all_arrangements(n):
    # 定义每个位置的可选取值
    value_options = [0, 1, None]
    # 生成长度为n的所有组合，repeat参数指定组合长度
    all_tuples = product(value_options, repeat=n)
    # 将元组转为列表格式匹配输出要求
    all_lists = [list(item) for item in all_tuples]
    # 若要和你给出的示例输出顺序一致（按None的数量升序排列），取消注释下面一行
    # all_lists.sort(key=lambda x: x.count(None))
    return {
        "noOfstates": len(all_lists),
        "states": all_lists
    }

# 测试n=3的场景
if __name__ == "__main__":
    result = generate_all_arrangements(3)
    print(f"总组合数：{result['noOfstates']}")
    for index, arrangement in enumerate(result["states"], 1):
        print(f"arrangement_{index} = {arrangement}")

结果说明 #

长度为n的列表总共有3^n种合法组合，比如n=3时总共有27种组合，你给出的示例只展示了前21种，剩余的混合取值组合以及全None的组合，上述代码都会完整返回。

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