Python动态生成可变范围嵌套循环及关联参数组合输出方案咨询

你之前对itertools的认知有误，itertools.product 本身是惰性迭代器，只有迭代到对应组合时才会生成该组合，不会提前加载所有组合到内存，完全满足你逐个输出喂给业务函数的需求。

基础纯Python实现 #

无需手动维护全局变量，新增参数仅需新增入参即可，自动关联参数名：

import itertools
from typing import Iterator, Dict, Tuple

def generate_param_combinations(*param_groups: Tuple[int, int, int, str]) -> Iterator[Dict[str, int]]:
    """
    传入任意多组 (start, stop, step, 参数名) 格式的参数组，返回组合迭代器
    """
    value_sequences = []
    param_names = []
    for start, stop, step, name in param_groups:
        value_sequences.append(range(start, stop, step))
        param_names.append(name)
    # 惰性生成组合，不会一次性加载所有数据
    for combo in itertools.product(*value_sequences):
        yield dict(zip(param_names, combo))

使用示例 #

# 模拟返回参数组的业务函数
def function_with_values():
    return 0, 3, 1, "param1"

def function_with_values_2():
    return 1, 4, 1, "param2"

# 迭代输出组合，直接通过参数名取值，新增参数无需修改取值逻辑
for combo in generate_param_combinations(function_with_values(), function_with_values_2()):
    print(combo["param1"], combo["param2"])

输出与你预期的结果完全一致：

0 1
0 2
0 3
1 1
1 2
1 3
2 1
2 2
2 3

如果需要新增第三组参数，仅需新增入参即可，无需修改函数内部逻辑：

def function_with_values_3():
    return 10, 13, 1, "param3"

for combo in generate_param_combinations(function_with_values(), function_with_values_2(), function_with_values_3()):
    print(combo["param1"], combo["param2"], combo["param3"])

Numpy优化版本 #

如果你需要配合numpy和numba做性能优化，仅需修改生成序列的部分即可：

import numpy as np
import itertools
from typing import Iterator, Dict, Tuple

def generate_numpy_param_combinations(*param_groups: Tuple[int, int, int, str]) -> Iterator[Dict[str, int]]:
    value_iters = []
    param_names = []
    for start, stop, step, name in param_groups:
        value_iters.append(np.nditer(np.arange(start, stop, step)))
        param_names.append(name)
    for combo in itertools.product(*value_iters):
        # 按需转换为Python原生int，不需要可直接返回numpy数值类型
        yield dict(zip(param_names, [int(v) for v in combo]))

该方案的核心优势：

• 内存占用极低，完全惰性迭代，适合大数量级的组合生成场景
• 扩展性极强，新增参数组不需要修改现有业务逻辑
• 性能天花板高，可无缝对接numba做进一步的运算加速

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