Python：循环可变数量的输入

Posted 2023-02-25

【中文标题】Python：循环可变数量的输入

【英文标题】：Python: Looping over a variable number of inputs

【发布时间】：2020-10-28 20:02:07

【问题描述】：

当可迭代项的数量根据用户输入而变化时，有没有一种方法可以有效地将可变数量的输入输入到像 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？谢谢！

【参考方案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)

【讨论】：

谢谢，效果很好！我发现，如果没有圆形命令，有时我会得到与整数略有偏差的数字。

【参考方案2】：

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

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

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