基于条件的循环范围

Question

我想说的是：

if switch == True:
    for i in range(5):
        # stuff()
else:
    for i in range(10):
        # stuff()

还有什么更多的毕达通的方法吗？

Answer 1

将它与条件表达式结合在一起

for i in range(5 if switch else 10):
    # do something with i

您不需要在这里测试== True；if已经为您做了这件事。

更易读的方法是将结束值分隔到一个变量中：

end = 5 if switch else 10
for i in range(end):
    # do something with i

Answer 2

使用内联短格式的if/else

for i in range(5 if switch else 10):
    # stuff()