有条件地使用嵌套 for 循环的一部分

Question

我已经广泛搜索了这个答案，但似乎找不到答案。因此，我第一次在这里发布问题。

我有一个使用许多参数来执行计算的函数。根据用户输入，我想遍历部分（或全部）参数的可能值。如果我想遍历所有参数，我可能会这样做：

for i in range(low1,high1):
    for j in range(low2,high2):
        for k in range(low3,high3):
            for m in range(low4,high4):
                doFunction(i, j, k, m)

如果我只想迭代第一个和第四个参数，我可能会这样做：

for i in range(low1,high1):
    for m in range(low4,high4):
        doFunction(i, user_input_j, user_input_k, m)

我的实际代码有将近 15 个嵌套的 for 循环，其中包含 15 个不同的参数——每个参数都可以是可迭代的（或不可迭代的）。因此，对于我来说，使用我拥有的东西并为参数的每个组合是否可迭代编写一个独特的 for 循环块是不可扩展的。如果我这样做，我将拥有 2^15 个不同的代码块。

我可以这样做：

if use_static_j == True:
    low2  = -999
    high2 = -1000
for i in range(low1,high1):
    for j in range(low2,high2):
        for k in range(low3,high3):
            for m in range(low4,high4):
                j1 = j if use_static_j==False else user_input_j
                doFunction(i, j1, k, m)

我只是想知道是否有更好的方法。也许使用 filter()、map() 或列表理解......（我还没有足够清楚的理解）

Answer 1

按照 cmets 中的建议，您可以构建一个参数数组，然后使用数组中的每个值调用函数。构建数组的最简单方法是对定义每个参数范围的列表使用递归。在这段代码中，我假设了一个由start、stop 和scale 参数组成的元组列表（例如，列表中的第三个元素产生[3, 2.8, 2.6, 2.4, 2.2]）。要使用静态值，您可以使用元组 (static, static+1, 1)。

def build_param_array(ranges):
    r = ranges[0]
    if len(ranges) == 1:
        return [[p * r[2]] for p in range(r[0], r[1], -1 if r[1] < r[0] else 1)]
    res = []
    for p in range(r[0], r[1], -1 if r[1] < r[0] else 1):
        pa = build_param_array(ranges[1:])
        for a in pa:
            res.append([p * r[2]] + a)
    return res

# range = (start, stop, scale)
ranges = [(1, 5, 1),
          (0, 10, .1),
          (15, 10, .2)
          ]

params = build_param_array(ranges)

for p in params:
    doFunction(*p)