Python：遍历可变数量的输入

如何解决Python：遍历可变数量的输入

当可迭代项的数量根据用户输入而变化时，是否有一种方法可以将可变数量的输入有效地馈送到像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,10)+round(phase[2],10)==100:
            print(phase)
            
if len(elem_list) == 4:
    for phase in itertools.product(steps,10)+round(phase[3],10)==100:
            print(phase)
            
if len(elem_list) == 5:
    for phase in itertools.product(steps,10)+round(phase[4],10)==100:
            print(phase)

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

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

解决方法

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)

,

就像@DeepSpace提到的那样，repeat =是处理所有迭代项都相同的完美方法。听起来就像您想要的。

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

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