如何在Python中生成指定长度、元素取值0-q的所有配置列表

需求是生成所有长度为n的列表，每个元素的取值范围是0到q（q为正整数）。例如q=1时，生成所有二进制列表；n=2、q=1时输出[[0,0],[0,1],[1,0],[1,1]]。任意q下，输出规模为(q+1)^n，比如n=3、q=2时会有27个结果。

已实现q=1时的递归代码，但不知道如何修改以适配任意q，且保证效率。原代码如下：

def generateAllSpinConfigs(n,arr,l,i):
    if i == n: 
        l.append(arr[:]) 
        return

    arr[i] = 0
    generateAllSpinConfigs(n,arr,l,i+1)  
    arr[i] = 1
    generateAllSpinConfigs(n,arr,l,i+1)

    return l

n=2
l=[]  
arr=[None]*n
print(generateAllSpinConfigs(n,arr,l,0))

>>[[0,0],[0,1],[1,0],[1,1]]

一、修改递归代码适配任意q #

原代码只硬编码了0和1，改成循环遍历0到q的每个值就能通用，逻辑和原递归保持一致：

def generate_all_configs(n, q):
    def helper(arr, index):
        if index == n:
            result.append(arr.copy())
            return
        # 遍历0到q的所有取值，替换原有的固定0、1
        for val in range(q + 1):
            arr[index] = val
            helper(arr, index + 1)
    
    result = []
    helper([None] * n, 0)
    return result

# 测试示例
print(generate_all_configs(2, 1))  # 输出[[0,0],[0,1],[1,0],[1,1]]
print(len(generate_all_configs(3, 2)))  # 输出27

二、迭代实现（避免递归栈限制） #

当n很大时，递归可能触发栈溢出，迭代方法更稳定，逐层构建所有配置：

def generate_all_configs_iter(n, q):
    configs = [[]]
    for _ in range(n):
        temp = []
        for config in configs:
            for val in range(q + 1):
                temp.append(config + [val])
        configs = temp
    return configs

# 测试
print(generate_all_configs_iter(2, 1))

三、用标准库itertools实现（最简洁高效） #

Python的itertools.product可以直接生成笛卡尔积，底层是C实现，性能最优，代码也最简洁：

import itertools

def generate_all_configs_itertools(n, q):
    # 生成range(q+1)的n次笛卡尔积，再转成列表
    return [list(item) for item in itertools.product(range(q + 1), repeat=n)]

# 测试
print(generate_all_configs_itertools(2, 1))
print(len(generate_all_configs_itertools(3, 2)))  # 输出27

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