有额外的循环条件......基于条件?

时间:2016-03-16 01:00:33

标签: python while-loop

变量a可以使用任意数量的值。 a的值是while循环的额外预定义条件的数量。

这可以通过多个elif语句完成,但有更简洁的方法吗?

if a == 0:
    while condition_1:
        ...
elif a == 1:
    while condition_1 or condition_2:
        ...
elif a == 2:
    while condition_1 or condition_2 or condition_3:
        ...

3 个答案:

答案 0 :(得分:10)

使用switch语句执行其他语言操作的一般方法是创建包含每个案例的函数的字典:

conds = {
    0: lambda: condition_1,
    1: lambda: condition_1 or condition_2,
    2: lambda: condition_1 or condition_2 or condition_3
}

然后:

while conds[a]():
    # do stuff

通过使用lambdas(或条件特别复杂的命名函数),每次循环时都可以评估适当的条件,而不是在定义字典时评估一次。

在这个简单的情况下,您的a具有从0开始的连续整数值,您可以使用列表并保存一些输入。为了进一步简化,您可以根据前一个条件定义每个条件,因为您每次只是添加一个条件:

conds = [
     lambda: condition_1,
     lambda: conds[0]() or condition_2,
     lambda: conds[1]() or condition_3
]

或者,如Julien在评论中所建议的那样:

conds = [
     lambda: condition_1,
     lambda: condition_2,
     lambda: condition_3
]

while any(cond() for cond in conds[:a+1]):
    # do stuff

答案 1 :(得分:3)

你有没有试过这样的事情:

while (a >= 0 and condition_1) or (a >= 1 and condition_2) or (a >= 2 and condition_3) ...

答案 2 :(得分:2)

您可以定义要为while评估的函数:

def test(a):
    if a == 1:
        return condition1(...)
    elif a == 2:
        return condition2(...) or condition1(...)
    elif a == 3:
        return condition2(...) or condition1(...) or condition3(...)
    else:
        return False

# test(a) will check the conditions ... define additional arguments if you need them
while test(a):
    do_stuff

它确实还有精灵,但你不需要多次写while - 循环。