可以使用递归函数来实现这种复杂嵌套循环,如下所示:
def nested_loop(levels, funcs, args):
'''
levels: 一个整数列表,表示每个层级的循环次数
funcs: 一个函数列表,表示每个层级要调用的函数
args: 一个二维列表,表示每个层级要传递给函数的参数
'''
# 如果层级为1,就直接循环调用函数并返回结果
if len(levels) == 1:
results = []
for i in range(levels[0]):
result = funcs[0](*args[0][i])
results.append(result)
return results
# 否则,递归调用下一层级
results = []
for i in range(levels[0]):
result = nested_loop(levels[1:], funcs[1:], args[1:])
results.extend(result)
return results
这个函数接受三个参数:
函数首先检查层级是否为1。如果是,则直接循环调用函数并返回结果。否则,通过递归调用下一层级来实现嵌套循环。最终将返回所有结果的列表。
以下是一个示例程序,使用此函数调用两个函数并传递不同的参数:
def func1(x, y):
return x + y
def func2(x, y, z):
return x * y - z
levels = [2, 3]
funcs = [func1, func2]
args = [[(i, j) for j in range(3)] for i in range(2)] + [[(i, j, k) for k in range(4)] for j in range(3) for i in range(2)]
results = nested_loop(levels, funcs, args)
print(results) # 输出:[0, 1, 2, 2, 3, 4, -0, -1, -2, -2, -3, -4, 2, 3, 4,