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)

Python：循环可变数量的输入

Python: Looping over a variable number of inputs

python

for-loop

itertools