Python：循环可变数量的输入

Question

当可迭代的数量根据用户输入而变化时，有没有办法有效地将可变数量的输入输入到像 itertools.product() 这样的程序中？

例如，下面的代码运行，但需要为每个循环显式定义：

steps = np.linspace(0,100,21)

if len(elem_list) == 2:
    for phase in itertools.product(steps,steps):
        if round(round(phase[0],10)+round(phase[1],10),10)==100:
            print(phase)

if len(elem_list) == 3:
    for phase in itertools.product(steps,steps,steps):
        if round(round(phase[0],10)+round(phase[1],10)+round(phase[2],10),10)==100:
            print(phase)
            
if len(elem_list) == 4:
    for phase in itertools.product(steps,steps,steps,steps):
        if round(round(phase[0],10)+round(phase[1],10)+round(phase[2],10)+round(phase[3],10),10)==100:
            print(phase)
            
if len(elem_list) == 5:
    for phase in itertools.product(steps,steps,steps,steps,steps):
        if round(round(phase[0],10)+round(phase[1],10)+round(phase[2],10)+round(phase[3],10)+round(phase[4],10),10)==100:
            print(phase)

其中 elem_list 包含用户输入的可变数量的元素。

有什么方法可以更简洁地编写它，因此它通常可以应用于任何长度的 elem_list？ 谢谢！

Answer 1

itertools.product完全为此接受repeat参数。 将它与sum和map一起使用。

我不确定为什么你需要round （或round(..., 10) ），因为steps包括整数（所以对它们求和肯定会给出一个整数），但这是要走的路。

steps = np.linspace(0, 100, 21)

for phase in itertools.product(steps, repeat=len(elem_list)):
    if sum(map(round, phase)) == 100:
        print(phase)

Answer 2

正如@DeepSpace 提到的，当所有迭代项都相同时，repeat= 是处理的完美方式。 这听起来像你想要的。

在更复杂的情况下，您可以使用范围列表。 并且这个列表可以任意生成。

list_of_ranges = [range(5), (1, 3, 7), range(10), (True, False)]
itertools.product(*list_of_ranges)