在Python中,我使用itertools.product()函数为模拟生成输入参数.

我有一个测试功能,需要4个输入参数a1,a2,b1和b2.我使用下面的代码生成参数.例：

params = itertools.product(range(10,41,2), range(10,41,2), range(0, 2), range(5, 31, 5))

……这给了我3072个组合.不幸的是,某些组合在逻辑上毫无意义. E. g.如果a2大于a1,则测试结果是无用的,当b1等于0时,b2的值完全无关 – 所以测试这样的组合是没有意义的.

是否有可能限制或过滤笛卡尔积,只需手动执行并嵌套for循环？因为我的真实用例有超过4个参数,这就是为什么我喜欢itertools中笛卡尔积函数的便利性.

任何想法或替代品？
任何帮助表示感谢,谢谢.

解决方法:

如果您有许多参数,使用像python-constraint这样的模块的基于约束的方法可能更容易使用 – 让它努力确定哪些组合是有效的.

这看起来像

from constraint import Problem

prob = Problem()
prob.addVariables(["a1", "a2"], range(10,41,2))
prob.addVariable("b1", [0, 2])
prob.addVariable("b2", range(5, 31, 5))
prob.addConstraint(lambda a1, a2: a2 <= a1, ["a1", "a2"])
prob.addConstraint(lambda b1, b2: b1 != 0 or b2 == 5, ["b1", "b2"])

for params in prob.getSolutionIter():
    run_sim(**params)