时间戳

首页 > 解决方案 > Python：循环可变数量的输入

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,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？谢谢！

标签： pythonfor-loopitertools

解决方案

itertools.productrepeat完全为此接受一个论点。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)

推荐阅读