Python实现字典任意数量参数的for循环自动化方案问询

嘿，这个问题用Python标准库的itertools.product就能轻松搞定！它可以帮你自动生成所有可变参数的笛卡尔积组合，完全不用手动写多层嵌套的for循环，不管temp里有多少个参数都能适配。

我帮你重写了Trial函数，直接看代码：

import numpy as np
import itertools

def f(data, t):
    x = data['focus'] * data['integration'] * data['upper']
    return t * x

def Trial(t, temp, data):
    calcdata = {}
    counter = 0
    # 提取所有可变参数的名称和对应的可选值列表
    param_names = list(temp.keys())
    param_value_lists = [temp[name] for name in param_names]
    
    # 生成所有参数组合的笛卡尔积
    for param_combo in itertools.product(*param_value_lists):
        # 创建原始data的副本，避免修改外部传入的data字典
        working_data = data.copy()
        # 把当前组合的参数值更新到工作字典中
        for name, value in zip(param_names, param_combo):
            working_data[name] = value
        
        # 计算函数结果
        x_axis, y_axis = np.vstack((t, np.vectorize(f)(working_data, t)))
        
        # 把结果和参数存入calcdata
        calcdata[counter] = {
            'xaxis': x_axis,
            'yaxis': y_axis
        }
        # 追加当前使用的参数组合
        for name, value in zip(param_names, param_combo):
            calcdata[counter][name] = value
        
        counter += 1
    return calcdata

# 测试代码（修复了原代码中未初始化data的小问题）
data = {'focus': 100}
t = np.linspace(1, 10, 10)
temp = {'integration': [1,2], 'upper': [3,4], 'other': [5,6]}
calcdata = Trial(t, temp, data)

# 可选：打印部分结果验证
for idx in calcdata:
    print(f"测试用例 {idx}:")
    params = {k: calcdata[idx][k] for k in temp.keys()}
    print(f"  使用参数: {params}")
    print(f"  Y轴前3个值: {calcdata[idx]['yaxis'][1][:3]}")

关键逻辑解释：

• itertools.product(*param_value_lists)：这个函数会把你传入的所有参数值列表做笛卡尔积运算，生成所有可能的参数组合。比如你有3个各含2个元素的列表，它会生成8组组合，和你手动写3层for循环的效果完全一样，而且不管你加多少个参数，这一行代码都能处理。
• data.copy()：这里创建了原始data的副本，避免直接修改外部的data字典——如果直接修改的话，外部的data会被污染，可能影响后续的其他代码，用副本更安全。如果你的data里有嵌套的可变对象（比如字典里套字典），可以换成copy.deepcopy(data)。
• zip(param_names, param_combo)：把参数名称和当前组合里的对应值一一配对，动态更新到工作字典中，这样不管参数数量多少，都能自动完成赋值。

另外注意我修复了原测试代码里的一个小bug：原代码中直接给data['focus']赋值前没有初始化data字典，会触发NameError，现在改成了data = {'focus': 100}来初始化。

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